Update app.py

app.py

@@ -4,7 +4,7 @@ from PIL import Image
 
 # Load the exported fastai Learner
 learn = load_learner('model.pkl')
-labels = learn.dls.vocab  # e.g. ['bird', 'shark']
+labels = learn.dls.vocab  # ['bird', 'shark']
 
 # Prediction function
 def predict(img):
@@ -21,11 +21,10 @@ examples = ['shark.jpg', 'bird.jpg']
 # Launch the Gradio interface
 gr.Interface(
     fn=predict,
-    inputs=gr.Image(type="pil"),  # <-- removed shape
+    inputs=gr.Image(type="pil"),
     outputs=gr.Label(num_top_classes=2),
     title=title,
     description=description,
     examples=examples,
-    interpretation='default',
     allow_flagging='never'
 ).launch()