Update app.py

app.py

@@ -8,7 +8,7 @@ import io
 def predict_input_image(img):
     # Normalize the image by cropping (center crop)
     img = Image.open(io.BytesIO(img))
-    h, w = img.size
+    h, w,_ = img.size
     crop_start_x = (w - 224) // 2
     crop_start_y = (h - 224) // 2
     img = img.crop((crop_start_x, crop_start_y, crop_start_x + 224, crop_start_y + 224))