Update app.py

app.py

@@ -68,10 +68,12 @@ def model_search_interface(user_input):
         models = get_models_for_task(task)
         if not models:
             return f"No models found for task '{task}'.", []
-        return f"Task identified: {task}", models
+        table_data = [[m["Model Name"], m["Task"], m["Architecture"]] for m in models]
+        return f"Task identified: {task}", table_data
     except Exception as e:
         return f"❌ Error: {str(e)}", []
 
+
 # 🎨 Launch UI
 with gr.Blocks() as demo:
     gr.Markdown("### 🔍 HuggingFace Model Search by Task")