Spaces:
Configuration error
Configuration error
Fedir Zadniprovskyi
commited on
Commit
·
9c0d580
1
Parent(s):
dc4f25f
fix: models route returning non-asr models
Browse files- faster_whisper_server/main.py +4 -2
- tests/api_model_test.py +1 -1
faster_whisper_server/main.py
CHANGED
@@ -86,7 +86,7 @@ def health() -> Response:
|
|
86 |
|
87 |
@app.get("/v1/models")
|
88 |
def get_models() -> list[ModelObject]:
|
89 |
-
models = huggingface_hub.list_models(library="ctranslate2")
|
90 |
models = [
|
91 |
ModelObject(
|
92 |
id=model.id,
|
@@ -105,7 +105,9 @@ def get_models() -> list[ModelObject]:
|
|
105 |
def get_model(
|
106 |
model_name: Annotated[str, Path(example="Systran/faster-distil-whisper-large-v3")],
|
107 |
) -> ModelObject:
|
108 |
-
models = list(
|
|
|
|
|
109 |
if len(models) == 0:
|
110 |
raise HTTPException(status_code=404, detail="Model doesn't exists")
|
111 |
exact_match: ModelInfo | None = None
|
|
|
86 |
|
87 |
@app.get("/v1/models")
|
88 |
def get_models() -> list[ModelObject]:
|
89 |
+
models = huggingface_hub.list_models(library="ctranslate2", tags="automatic-speech-recognition")
|
90 |
models = [
|
91 |
ModelObject(
|
92 |
id=model.id,
|
|
|
105 |
def get_model(
|
106 |
model_name: Annotated[str, Path(example="Systran/faster-distil-whisper-large-v3")],
|
107 |
) -> ModelObject:
|
108 |
+
models = list(
|
109 |
+
huggingface_hub.list_models(model_name=model_name, library="ctranslate2", tags="automatic-speech-recognition")
|
110 |
+
)
|
111 |
if len(models) == 0:
|
112 |
raise HTTPException(status_code=404, detail="Model doesn't exists")
|
113 |
exact_match: ModelInfo | None = None
|
tests/api_model_test.py
CHANGED
@@ -4,7 +4,7 @@ from faster_whisper_server.server_models import ModelObject
|
|
4 |
|
5 |
MODEL_THAT_EXISTS = "Systran/faster-whisper-tiny.en"
|
6 |
MODEL_THAT_DOES_NOT_EXIST = "i-do-not-exist"
|
7 |
-
MIN_EXPECTED_NUMBER_OF_MODELS =
|
8 |
|
9 |
|
10 |
# HACK: because ModelObject(**data) doesn't work
|
|
|
4 |
|
5 |
MODEL_THAT_EXISTS = "Systran/faster-whisper-tiny.en"
|
6 |
MODEL_THAT_DOES_NOT_EXIST = "i-do-not-exist"
|
7 |
+
MIN_EXPECTED_NUMBER_OF_MODELS = 70 # At the time of the test creation there are 89 models
|
8 |
|
9 |
|
10 |
# HACK: because ModelObject(**data) doesn't work
|