Update app.py

app.py

@@ -1,5 +1,11 @@
 import gradio as gr
 from transformers import AutoTokenizer
+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):
@@ -11,13 +17,18 @@ def count_tokens(llm_name, input_text):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# List of supported transformer-based models
-llm_options = [
-    "gpt2", 
-    "EleutherAI/gpt-neo-2.7B", 
-    "bigscience/bloom-560m", 
-    "facebook/opt-1.3b"
-]
+# 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
+    })
+
+# Sort models by number of downloads or likes
+llm_options = sorted(model_details, key=lambda x: x['downloads'], reverse=True)
 
 # Define custom CSS for a bluish theme
 custom_css = """