clickable links

app.py

@@ -53,9 +53,6 @@ def get_model_stats(search_term):
     # Read the CSV file into a pandas DataFrame
     df = pd.read_csv(output_file)
     
-    # Create clickable links for model IDs
-    df['Model ID'] = df['Model ID'].apply(lambda x: f'<a href="https://huggingface.co/{x}" target="_blank">{x}</a>')
-    
     # Create status message with total downloads
     status_message = (
         f"Found {model_count} models for search term '{search_term}'\n"
@@ -66,6 +63,9 @@ 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")
@@ -95,10 +95,19 @@ with gr.Blocks(title="Hugging Face Model Statistics") as demo:
     # Store the CSV file path in a state
     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)
+        return df, status, csv_path
+    
     search_button.click(
         fn=get_model_stats,
         inputs=search_input,
         outputs=[output_table, status_message, csv_path]
+    ).then(
+        fn=process_results,
+        inputs=[output_table, status_message, csv_path],
+        outputs=[output_table, status_message, csv_path]
     )
     
     download_button.click(