refactor

app.py

@@ -13,8 +13,9 @@ def classify_image(image):
     pred, idx, probs = learner.predict(image)
     return dict(zip(categories, map(float, probs)))
 
-
+image = gr.Image(shape=(192, 192))
+label = gr.Label()
 examples = ['chess.jpg', 'checkers.webp', "dono.jpg"]
 
-demo = gr.Interface(fn=classify_image, inputs="image", outputs='label', examples=examples)
+demo = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
 demo.launch()