Update app.py

app.py

@@ -19,37 +19,14 @@ def load_and_preprocess_image(img, target_size=(256, 256)):
 
 # Step 4: Function to make predictions
 def predict_image(img):
-    print("Image received.")  # Confirm function is triggered
-
-    try:
-        img_array = load_and_preprocess_image(img)
-        print("Image shape:", img_array.shape)
-
-        prediction = model.predict(img_array)
-        print("Raw prediction:", prediction)
-
-        # Determine output shape
-        if prediction.shape[-1] == 1:
-            prob_real = prediction[0][0]
-        elif prediction.shape[-1] == 2:
-            prob_real = prediction[0][1]  # assumes index 1 = "real"
-        else:
-            print("Unexpected model output shape.")
-            return "Error: Unrecognized output shape.", ""
-
-        real_confidence = prob_real * 100
-        fake_confidence = (1 - prob_real) * 100
-        result_label = "Real" if real_confidence > fake_confidence else "Fake"
-
-        result_text = f"The model predicts this image is '{result_label}' with {max(real_confidence, fake_confidence):.2f}% confidence."
-        explanation = f"Real Confidence: {real_confidence:.2f}% | Fake Confidence: {fake_confidence:.2f}%"
-
-        print("Prediction result:", result_text)
-        return result_text, explanation
-
-    except Exception as e:
-        print("Error during prediction:", str(e))
-        return "An error occurred during prediction.", str(e)
+    img_array = load_and_preprocess_image(img)
+    prediction = model.predict(img_array)[0][0]
+    real_confidence = prediction * 100
+    fake_confidence = (1 - prediction) * 100
+    result_label = "Real" if real_confidence > fake_confidence else "Fake"
+    result_text = f"The model predicts this image is '{result_label}' with {max(real_confidence, fake_confidence):.2f}% confidence."
+    explanation = f"Real Confidence: {real_confidence:.2f}% | Fake Confidence: {fake_confidence:.2f}%"
+    return result_text, explanation
 
 # Step 5: Define the Gradio interface
 interface = gr.Interface(
@@ -66,4 +43,3 @@ interface = gr.Interface(
 # Step 6: Launch the app
 if __name__ == "__main__":
     interface.launch()
-