Update app.py

app.py

@@ -59,13 +59,16 @@ def generate_text(message, history):
 
 
 def predict(img):
-    img = PILImage.create(img)
+    try:
+        img = PILImage.create(img)
+    except:
+        return {"bird": "Unknown"}
     pred,pred_idx,probs = learn.predict(img)
     return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
 title = "Bird Detector"
 description = "Bird Detector."
-examples = ['BIRD.png']
+examples = ['BIRD.jpg']
 interpretation='default'
 enable_queue=True
 
@@ -78,6 +81,4 @@ def combined(img, message):
     return response
 
 
-
-
-gr.Interface(fn=combined,inputs=Image(shape=(512, 512)),outputs=Text(),title=title,description=description,examples=examples,interpretation=interpretation,enable_queue=enable_queue).launch()
+gr.Interface(fn=combined,inputs=Image(shape=(512, 512)),outputs=Text(),title=title,description=description,examples=examples,interpretation=interpretation,enable_queue=enable_queue).launch()