Update app.py

app.py

@@ -17,14 +17,12 @@ def predict(image):
     predictions_v1 = model_pipeline_v1(resized_image)
     predictions_v2 = model_pipeline_v2(resized_image)
     
-    # Format the results for display
+    # Format the results for each model
     results_v1 = {p["label"]: p["score"] for p in predictions_v1}
     results_v2 = {p["label"]: p["score"] for p in predictions_v2}
     
-    return {
-        "Model v1 Predictions": results_v1,
-        "Model v2 Predictions": results_v2,
-    }
+    # Return results as separate outputs
+    return results_v1, results_v2
 
 # Define the Gradio Interface
 gr.Interface(