Update app.py

app.py

@@ -25,19 +25,27 @@ def predict_image(image):
     pred1 = model1.predict(preprocessed_image)[0][0]
     pred2 = model2.predict(preprocessed_image)[0][0]
     
-    # Average the predictions
-    avg_pred = (pred1 + pred2) / 2
-    result = "Real" if avg_pred > 0.5 else "Fake"
-    confidence = avg_pred if avg_pred > 0.5 else 1 - avg_pred
+    # Prepare results for each model
+    result1 = "Real" if pred1 > 0.5 else "Fake"
+    confidence1 = pred1 if pred1 > 0.5 else 1 - pred1
     
-    return f"{result} (Confidence: {confidence:.2f})"
+    result2 = "Real" if pred2 > 0.5 else "Fake"
+    confidence2 = pred2 if pred2 > 0.5 else 1 - pred2
+    
+    return (
+        f"Model 1: {result1} (Confidence: {confidence1:.2f})",
+        f"Model 2: {result2} (Confidence: {confidence2:.2f})"
+    )
 
 iface = gr.Interface(
     fn=predict_image,
     inputs=gr.Image(),
-    outputs="text",
+    outputs=[
+        gr.Textbox(label="Model 1 Prediction"),
+        gr.Textbox(label="Model 2 Prediction")
+    ],
     title="Real vs Fake Face Detection",
-    description="Upload an image to determine if it's a real or fake face."
+    description="Upload an image to determine if it's a real or fake face using two different models."
 )
 
 iface.launch()