Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -53,6 +53,8 @@ def get_prompt(name):
|
|
53 |
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:
|
|
|
53 |
def fetch_models(task):
|
54 |
"""Fetch models for a specific task from Hugging Face Hub."""
|
55 |
try:
|
56 |
+
all_models=list_models()
|
57 |
+
print(all_models)
|
58 |
models = list_models(filter=f"pipeline_tags:{task}")
|
59 |
return [model.modelId for model in models]
|
60 |
except Exception as e:
|