K00B404 commited on
Commit
8d65776
·
verified ·
1 Parent(s): f42689b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
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)}"]