Update app.py

app.py

@@ -13,7 +13,10 @@ def predict_and_visualize(img):
     # Store the original image size
     original_size = img.size
     
-
+    # Convert the input image to the target size expected by the model
+    img_resized = img.resize((224,224))
+    img_array = np.array(img_resized) / 255.0  # Normalize the image
+    img_array = np.expand_dims(img_array, axis=0)  # Add batch dimension
 
     # Make a prediction
     prediction = model.predict(img_array)
@@ -39,4 +42,4 @@ iface = gr.Interface(
 )
 
 # Launch the Gradio app
-iface.launch()
+iface.launch()