Update app.py

app.py

@@ -8,6 +8,8 @@ labels = learn.dls.vocab
 
 def predict(img):
     img = PILImage.create(img)
+    size = (224, 224)
+    img = img.thumbnail(size, Image.ANTIALIAS)
     _pred, _pred_w_idx, probs = learn.predict(img)
     # gradio doesn't support tensors, so converting to float
     labels_probs = {labels[i]: float(probs[i]) for i, _ in enumerate(labels)}
@@ -28,7 +30,7 @@ interface_options = {
 demo = gr.Interface(
     fn=predict,
     inputs=gr.inputs.Image(shape=(480, 480)),
-    outputs=gr.outputs.Label(num_top_classes=3),
+    outputs=gr.outputs.Label(num_top_classes=2),
     **interface_options,
 )