30 days

app.py

@@ -1,10 +1,9 @@
-from huggingface_hub import HfApi
+import tempfile
 import csv
-import os
-import gradio as gr
 import pandas as pd
+import gradio as gr
+from huggingface_hub import HfApi
 from pathlib import Path
-import tempfile
 
 def get_model_stats(search_term):
     # Initialize the API
@@ -15,10 +14,10 @@ def get_model_stats(search_term):
     output_file = Path(temp_dir) / f"{search_term}_models_alltime.csv"
     
     # Get the generator of models with the working sort parameter
-    print(f"Fetching {search_term} models with all-time download statistics...")
+    print(f"Fetching {search_term} models with download statistics...")
     models_generator = api.list_models(
         search=search_term, 
-        expand=["downloadsAllTime"],
+        expand=["downloads", "downloadsAllTime"],  # Get both 30-day and all-time downloads
         sort="_id"  # Sort by ID to avoid timeout issues
     )
     
@@ -26,7 +25,7 @@ def get_model_stats(search_term):
     with open(output_file, 'w', newline='', encoding='utf-8') as csvfile:
         csv_writer = csv.writer(csvfile)
         # Write header
-        csv_writer.writerow(["Model ID", "Downloads (30 days)", "Downloads (All Time)", "Trending Score"])
+        csv_writer.writerow(["Model ID", "Downloads (30 days)", "Downloads (All Time)"])
         
         # Process models
         model_count = 0
@@ -35,8 +34,7 @@ def get_model_stats(search_term):
             csv_writer.writerow([
                 getattr(model, 'id', "Unknown"),
                 getattr(model, 'downloads', 0),  # Last 30 days downloads
-                getattr(model, 'downloads_all_time', 0),
-                getattr(model, 'trending_score', 0)
+                getattr(model, 'downloads_all_time', 0)
             ])
             model_count += 1
     
@@ -61,8 +59,8 @@ with gr.Blocks(title="Hugging Face Model Statistics") as demo:
     
     with gr.Row():
         output_table = gr.Dataframe(
-            headers=["Model ID", "Downloads (30 days)", "Downloads (All Time)", "Trending Score"],
-            datatype=["str", "number", "number", "number"],
+            headers=["Model ID", "Downloads (30 days)", "Downloads (All Time)"],
+            datatype=["str", "number", "number"],
             label="Model Statistics"
         )
         status_message = gr.Textbox(label="Status")
@@ -87,4 +85,4 @@ with gr.Blocks(title="Hugging Face Model Statistics") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()