Add /models API endpoint to list available detection models

app.py

@@ -397,6 +397,15 @@ with gr.Blocks(title="Industrial Defect Detection") as demo:
     - Tire Cord
     """)
 
+# Add custom endpoint for listing models
+@demo.api_endpoint("/models", method="GET")
+def list_models():
+    """API endpoint to list available models."""
+    return {
+        "models": [model_info["name"] for model_info in MODELS.values()],
+        "count": len(MODELS)
+    }
+
 # Launch the app
 if __name__ == "__main__":
     demo.launch()