Update app.py

app.py

@@ -8,13 +8,12 @@ import os
 model = load_model(r'deepfake_detection_mobilenet_model.h5')
 
 def predict_image(img):
-    
     x = img_to_array(img)
-
-    x = cv2.resize(x, (256, 256), interpolation=cv2.INTER_AREA)
+    
+    # Resize to the expected input size of the model
+    x = cv2.resize(x, (224, 224), interpolation=cv2.INTER_AREA)
     
     x /= 255.0
-
     x = np.expand_dims(x, axis=0)
 
     prediction = np.argmax(model.predict(x), axis=1)
@@ -24,15 +23,12 @@ def predict_image(img):
     else:
         return 'Real Image'
 
-
 # Define the Gradio Interface with the desired title and description
 
 description_html = """
 <p>Upload a face image to check if it's real or morphed with deepfake</p>
 """
 
-# Define example images and their true labels for users to choose from
-
 custom_css = """
 div {background-color: whitesmoke;}
 """
@@ -44,4 +40,4 @@ gr.Interface(
     title="Deepfake Image Detection",
     description=description_html,
     allow_flagging='never'
-).launch()
+).launch()