yes

app.py

@@ -10,12 +10,15 @@ import gradio as gr
 
 learn = load_learner(Path("model.pkl"))
 
-categories = {'Pasta', 'Pizza'}
+# categories = {'Pizza', 'Pasta'}
 
 def classify_image(img):
     pred, idx, probs = learn.predict(img)
-    print(dict(zip(categories, map(float, probs))))
-    return dict(zip(categories, map(float, probs)))
+    # print(pred)
+    # print(idx)
+    # print(dict(zip(categories, map(float, probs))))
+    return f"This is {pred}.\nConfidence Level: {float(max(probs))}"
+    # return dict(zip(categories, map(float, probs)))
 
 
 image = gr.Image(shape=(192, 192))
@@ -24,6 +27,6 @@ examples = ['pizza.jpg', 'pasta.jpg', 'dunno.jpg']
 
 app = gr.Interface(fn=classify_image,
                    inputs=image,
-                   outputs=label,
+                   outputs=gr.Textbox(label="Output"),
                    examples=examples)
 app.launch()