Update app.py

app.py

@@ -36,7 +36,7 @@ cap_labels = (
      'visor cap'
 )
 
-model = load_learner('cap-recognizer-v0.pkl')
+model = load_learner('models/cap-recognizer-v0.pkl')
 
 def recognize_image(image):
     pred, idx, probs = model.predict(image)
@@ -50,5 +50,5 @@ examples = [
     'unknown02.png'
     ]
 
-iface = gr.Interface(recognize_image, gr.Image(shape=(192,192)), gr.Label(num_top_classes=5), examples=examples)
+iface = gr.Interface(recognize_image, gr.Image(), gr.Label(), examples=examples)
 iface.launch(inline=False)