Update app.py

app.py

@@ -6,21 +6,24 @@ import pandas as pd
 # Initialize the Hugging Face API
 api = HfApi()
 
-def get_recent_models(min_likes, days_ago):
+def get_recent_models(min_likes, days_ago, filter_substrings, search_substrings):
     # Get the current date and date from `days_ago` days ago
     today = datetime.utcnow().replace(tzinfo=None)
     start_date = (today - timedelta(days=days_ago)).replace(tzinfo=None)
     
     # Initialize an empty list to store the filtered models
     recent_models = []
+
+    filter_substrings = [sub.strip().lower() for sub in filter_substrings.split(';')]
+    search_substrings = [term.strip().lower() for term in search_substrings.split(';')]
     
     # Use a generator to fetch models in batches, sorted by likes in descending order
     for model in api.list_models(sort="likes", direction=-1):
-        if model.likes > 10:
+        if model.likes >= min_likes:
             if hasattr(model, "created_at") and model.created_at:
                 # 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:
+                if created_at_date >= start_date and not any(sub in model.modelId.lower() for sub in filter_substrings) and any(term in model.modelId.lower() for term in search_substrings):
                     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>',
@@ -41,10 +44,13 @@ 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("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. Use `;` to split filter and search")
+    with gr.Row():
+        likes_slider = gr.Slider(minimum=1, maximum=100, step=1, value=5, label="Minimum Likes")
+        days_slider = gr.Slider(minimum=1, maximum=30, step=1, value=3, label="Days Ago")
     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")
+        filter_text = gr.Text(label="Filter", max_lines=1)
+        search_text = gr.Text(label="Search", max_lines=1)
     
     btn = gr.Button("Run")
     
@@ -55,7 +61,7 @@ with gr.Blocks() as demo:
             datatype=["html", "number", "str"],
         )
     
-    btn.click(fn=get_recent_models, inputs=[likes_slider, days_slider], outputs=df)
+    btn.click(fn=get_recent_models, inputs=[likes_slider, days_slider, filter_text, search_text], outputs=df)
 
 if __name__ == "__main__":
     demo.launch()