K00B404 commited on
Commit
2c59fff
·
verified ·
1 Parent(s): 3f66775

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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: