Update app.py

app.py

@@ -10,7 +10,9 @@ labels = learn.dls.vocab
 
 # Prediction function
 def predict(img):
+    # Resize the image inside the function (if required)
     img = PILImage.create(img)
+    img = img.resize((512, 512))  # Resize image to (512, 512)
     pred, pred_idx, probs = learn.predict(img)
     prediction = str(pred)
     return prediction
@@ -33,8 +35,8 @@ enable_queue = True
 # Update the interface components
 gr.Interface(
     fn=predict,
-    inputs=gr.Image(shape=(512, 512)),  # Updated input component
-    outputs=gr.Label(num_top_classes=3),  # Updated output component
+    inputs=gr.Image(type="pil"),  # Use the 'type' argument instead of 'shape'
+    outputs=gr.Label(num_top_classes=3),
     title=title,
     description=description,
     article=article,