Update app.py

app.py

@@ -4,9 +4,6 @@ from huggingface_hub import HfApi
 
 api = HfApi()
 
-# List all models
-models = list(api.list_models())
-
 # Define a function to calculate tokens
 def count_tokens(llm_name, input_text):
     try:
@@ -18,17 +15,10 @@ def count_tokens(llm_name, input_text):
         return f"Error: {str(e)}"
 
 # Fetch model details including metadata (like download count, likes)
-model_details = []
-for model in models:
-    details = api.model_info(model.modelId)
-    model_details.append({
-        'modelId': model.modelId,
-        'downloads': details.downloads,  # or any other relevant metric
-        'likes': details.likes           # or likes
-    })
+models = list(api.list_models(task="text-generation"))
 
 # Sort models by number of downloads or likes
-llm_options = sorted(model_details, key=lambda x: x['downloads'], reverse=True)
+llm_options = models#sorted(model_details, key=lambda x: x['downloads'], reverse=True)
 
 # Define custom CSS for a bluish theme
 custom_css = """