Update app.py

app.py

@@ -93,12 +93,15 @@ def display_table(search_query="", filters=[], use_cache=False):
         filtered_df = df
 
     if filters:
-        if "Free" in filters:
-            filtered_df = filtered_df[filtered_df["API"] == "Free"]
+        api_filters = [f for f in filters if f in ["Free", "Pro Subscription", "Not Responding"]]
+        if api_filters:
+            filtered_df = filtered_df[filtered_df["API"].isin(api_filters)]
         if "Text Completion" in filters:
             filtered_df = filtered_df[filtered_df["Text Completion"] == "✓"]
         if "Chat Completion" in filters:
             filtered_df = filtered_df[filtered_df["Chat Completion"] == "✓"]
+        if "Vision" in filters:
+            filtered_df = filtered_df[filtered_df["Vision"] == "✓"]
 
     styled_df = filtered_df.style.apply(apply_row_styles, axis=1, subset=["Model", "API", "Text Completion", "Chat Completion", "Vision"])
     return styled_df
@@ -157,7 +160,7 @@ with gr.Blocks() as demo:
     gr.Markdown("## HF Serverless LLM Inference API Status")
     gr.Markdown(description)
     search_box = gr.Textbox(label="Search for a model", placeholder="Type model name here...")
-    filter_box = gr.CheckboxGroup(choices=["Free", "Text Completion", "Chat Completion"], label="Filters")
+    filter_box = gr.CheckboxGroup(choices=["Free", "Pro Subscription", "Not Responding", "Text Completion", "Chat Completion", "Vision"], label="Filters")
     table = gr.Dataframe(value=display_table(use_cache=True), headers="keys")
 
     def update_filters(query, filters):