Get all models summary

Dockerfile

@@ -0,0 +1,8 @@
+FROM python:3.9-slim
+
+# Upgrade pip
+RUN pip install --upgrade pip
+
+# Install dependencies from requirements.txt
+COPY requirements.txt /tmp/requirements.txt
+RUN pip install --no-cache-dir -r /tmp/requirements.txt

app.py

@@ -0,0 +1,52 @@
+import gradio as gr
+from huggingface_hub import HfApi
+
+def fetch_model_info(api_token):
+    # Initialize API
+    api = HfApi()
+    
+    # Authenticate using the provided token
+    user = api.whoami(api_token)
+    username = user["name"]
+    
+    # Get models accessible to the user
+    models = api.list_models(author=username)
+    
+    # Prepare data for the table
+    data = []
+    for model in models:
+        model_name = model.modelId
+        model_version = model.cardData.get("library_name", "Unknown")
+        parameter_count = model.cardData.get("parameters", "Unknown")
+        model_path = f"https://huggingface.co/{model_name}"
+        data.append([model_name, model_version, parameter_count, model_path])
+    
+    return data
+
+def display_models(api_token):
+    try:
+        # Fetch model info
+        models_data = fetch_model_info(api_token)
+        
+        # Create HTML table
+        table_html = "<table border='1' style='border-collapse:collapse;width:100%;'>"
+        table_html += "<tr><th>Model Name</th><th>Version</th><th>Parameter Count</th><th>Path</th></tr>"
+        for row in models_data:
+            table_html += f"<tr><td>{row[0]}</td><td>{row[1]}</td><td>{row[2]}</td><td><a href='{row[3]}' target='_blank'>Link</a></td></tr>"
+        table_html += "</table>"
+        
+        return table_html
+    except Exception as e:
+        return f"Error fetching models: {e}"
+
+# Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("### Hugging Face Model Information")
+    api_token_input = gr.Textbox(label="Hugging Face API Token", type="password")
+    output_html = gr.HTML()
+    fetch_button = gr.Button("Fetch Models")
+    
+    fetch_button.click(display_models, inputs=[api_token_input], outputs=[output_html])
+
+# Launch the app
+demo.launch()

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+huggingface_hub