Update app.py

app.py

@@ -245,7 +245,7 @@ def predict(image):
     input_shape = model.get_inputs()[0].shape
 
     # Resize the image to the model's input shape
-    image = cv2.resize(image, (input_shape[2], input_shape[3]))
+    image = cv2.resize(image, (640, 640))
 
     # Convert the image to a numpy array and add a batch dimension
     image = np.expand_dims(image, axis=0)