whoops

app.py

@@ -10,8 +10,7 @@ import gradio as gr
 
 learn = load_learner(Path("model.pkl"))
 
-categories = {'Pizza', 'Pasta'}
-
+categories = {'Pasta', 'Pizza'}
 
 def classify_image(img):
     pred, idx, probs = learn.predict(img)
@@ -22,8 +21,8 @@ image = gr.Image(shape=(192, 192))
 label = gr.Label()
 examples = ['pizza.jpg', 'pasta.jpg', 'dunno.jpg']
 
-app = gr.Interface(fn=classify_image, 
+app = gr.Interface(fn=classify_image,
                    inputs=image,
-                   outputs=label, 
+                   outputs=label,
                    examples=examples)
 app.launch()