links

app.py

@@ -63,9 +63,6 @@ def get_model_stats(search_term):
     # Return both the DataFrame, status message, and the CSV file path
     return df, status_message, str(output_file)
 
-def format_model_link(model_id):
-    return f'<a href="https://huggingface.co/{model_id}" target="_blank">{model_id}</a>'
-
 # Create the Gradio interface
 with gr.Blocks(title="Hugging Face Model Statistics") as demo:
     gr.Markdown("# Hugging Face Model Statistics")
@@ -80,13 +77,14 @@ with gr.Blocks(title="Hugging Face Model Statistics") as demo:
         search_button = gr.Button("Search")
     
     with gr.Row():
-        output_table = gr.Dataframe(
-            headers=["Model ID", "Downloads (30 days)", "Downloads (All Time)"],
-            datatype=["str", "number", "number"],
-            label="Model Statistics",
-            wrap=True
-        )
-        status_message = gr.Textbox(label="Status", lines=3)  # Increased lines to show all stats
+        with gr.Column():
+            output_table = gr.Dataframe(
+                headers=["Model ID", "Downloads (30 days)", "Downloads (All Time)"],
+                datatype=["str", "number", "number"],
+                label="Model Statistics",
+                wrap=True
+            )
+            status_message = gr.Textbox(label="Status", lines=3)
     
     with gr.Row():
         download_button = gr.Button("Download CSV")
@@ -96,8 +94,14 @@ with gr.Blocks(title="Hugging Face Model Statistics") as demo:
     csv_path = gr.State()
     
     def process_results(df, status, csv_path):
-        # Format the model IDs as clickable links
-        df['Model ID'] = df['Model ID'].apply(format_model_link)
+        # Create HTML links for each model
+        html_links = []
+        for model_id in df['Model ID']:
+            html_links.append(f'<a href="https://huggingface.co/{model_id}" target="_blank">{model_id}</a>')
+        
+        # Update the DataFrame with HTML links
+        df['Model ID'] = html_links
+        
         return df, status, csv_path
     
     search_button.click(