model type

app.py

@@ -21,10 +21,12 @@ def get_recent_models(min_likes, days_ago):
                 # Ensure created_at is offset-naive
                 created_at_date = model.created_at.replace(tzinfo=None)
                 if created_at_date >= start_date and model.likes >= min_likes:
+                    task = model.pipeline_tag if hasattr(model, "pipeline_tag") else "N/A"
                     recent_models.append({
                         "Model ID": f'<a href="https://huggingface.co/{model.modelId}" target="_blank">{model.modelId}</a>',
                         "Likes": model.likes,
-                        "Creation Date": model.created_at.strftime("%Y-%m-%d %H:%M")
+                        "Creation Date": model.created_at.strftime("%Y-%m-%d %H:%M"),
+                        "Task": task
                     })
         else:
             # Since the models are sorted by likes in descending order,
@@ -38,8 +40,9 @@ def get_recent_models(min_likes, days_ago):
 
 # Define the Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Model Drops Tracker 🚀")
-    gr.Markdown("Overwhelmed by the rapid pace of model releases? 😅 You're not alone! That's exactly why I built this tool. Easily filter recent models from the Hub by setting a minimum number of likes and the number of days since their release. Click on a model to see its card.")
+    gr.Markdown("# Recently Created Models on Hugging Face")
+    gr.Markdown("Filter recent models from the Hugging Face Hub based on minimum likes and the number of days ago they were created.")
+    
     with gr.Row():
         likes_slider = gr.Slider(minimum=0, maximum=100, step=10, value=10, label="Minimum Likes")
         days_slider = gr.Slider(minimum=1, maximum=7, step=1, value=1, label="Days Ago")
@@ -48,9 +51,10 @@ with gr.Blocks() as demo:
     
     with gr.Column():
         df = gr.DataFrame(
-            headers=["Model ID", "Likes", "Creation Date"],
+            headers=["Model ID", "Likes", "Creation Date", "Task"],
             wrap=True,
-            datatype=["html", "number", "str"],
+            datatype=["html", "number", "str", "str"],
+            interactive=True
         )
     
     btn.click(fn=get_recent_models, inputs=[likes_slider, days_slider], outputs=df)