chore: Add fuzzywuzzy dependency and implement model search functionality

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import pandas as pd
+from fuzzywuzzy import fuzz
 
 def load_leaderboard():
     imagenet_df = pd.read_csv('https://raw.githubusercontent.com/huggingface/pytorch-image-models/main/results/results-imagenet.csv')
@@ -51,13 +52,40 @@ def load_leaderboard():
 
     return result
 
+global df
 df = load_leaderboard()
 
+def search_leaderboard(model_name):
+    if not model_name:
+        return df
+    
+    threshold = 95  # You can adjust this value to make the search more or less strict
+
+    def calculate_similarity(row):
+        similarity = fuzz.partial_ratio(model_name.lower(), row['model'].lower())
+        return similarity if similarity >= threshold else 0
+
+    # Add a new column for similarity scores
+    df['similarity'] = df.apply(calculate_similarity, axis=1)
+
+    # Filter and sort the dataframe
+    filtered_df = df[df['similarity'] > 0].sort_values('similarity', ascending=False)
+
+    # Remove the similarity column before returning
+    filtered_df = filtered_df.drop('similarity', axis=1)
+
+    return filtered_df
+
 with gr.Blocks("Timm Leaderboard") as app:
     gr.HTML("<center><h1>PyTorch Image Models Leaderboard</h1></center>")
     gr.Markdown("This leaderboard is based on the results of the models from the [PyTorch Image Models]('https://github.com/huggingface/pytorch-image-models') repository.")
-    leaderboard = gr.DataFrame(df)
+    with gr.Row():
+        search_bar = gr.Textbox(lines=1, label="Search Model (You can press Enter to Search)", placeholder="Search for a model", scale=4)
+        search_btn = gr.Button(value="Search", variant="primary", scale=1)
+    leaderboard = gr.Dataframe(df)
     refresh_button = gr.Button(value="Refresh Leaderboard", variant="primary")
     refresh_button.click(load_leaderboard, outputs=[leaderboard])
+    search_btn.click(search_leaderboard, inputs=[search_bar], outputs=[leaderboard])
+    search_bar.submit(search_leaderboard, inputs=[search_bar], outputs=[leaderboard])
 
 app.launch()

requirements.txt

@@ -1,2 +1,3 @@
 gradio
-pandas
+pandas
+fuzzywuzzy