Update app.py

app.py

@@ -25,7 +25,7 @@ model = create_mock_model()
 
 # Function to preprocess the image and make predictions
 def predict_image(image):
-    # Resize the image to 224x224 and normalize
+    # Resize and normalize the image
     img = cv2.resize(image, (224, 224))
     img = img / 255.0  # Normalize to [0, 1] range
 
@@ -36,8 +36,8 @@ def predict_image(image):
     label = train_info[top_class]  # Get the corresponding label
     return label
 
-# Updated Gradio interface
-input_image = gr.Image(shape=(224, 224))
+# Define the Gradio interface without the 'shape' argument
+input_image = gr.Image()  # Removed 'shape' argument
 output_label = gr.Label()
 
 # Launch the Gradio interface