Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -54,6 +54,12 @@ def fetch_models(task):
|
|
54 |
"""Fetch models for a specific task from Hugging Face Hub."""
|
55 |
try:
|
56 |
models = list_models(filter=f"pipeline_tags:{task}")
|
|
|
|
|
|
|
|
|
|
|
|
|
57 |
return [model.modelId for model in models]
|
58 |
except Exception as e:
|
59 |
return [f"Error fetching models: {str(e)}"]
|
|
|
54 |
"""Fetch models for a specific task from Hugging Face Hub."""
|
55 |
try:
|
56 |
models = list_models(filter=f"pipeline_tags:{task}")
|
57 |
+
models += [
|
58 |
+
"HuggingFaceH4/zephyr-7b-beta",
|
59 |
+
"HuggingFaceH4/zephyr-7b-alpha",
|
60 |
+
"HuggingFaceH4/zephyr-6b"
|
61 |
+
]
|
62 |
+
|
63 |
return [model.modelId for model in models]
|
64 |
except Exception as e:
|
65 |
return [f"Error fetching models: {str(e)}"]
|