Formatting output

app.py

@@ -6,7 +6,7 @@ categories = ('Beaver Dam', 'Not a Beaver Dam')
 
 def is_it_a_beaver_dam(input_img):
     pred, idx, probs = learner.predict(input_img)
-    return {dict(zip(categories, map(float, probs)))}
+    return f'{pred} {dict(zip(categories, map(float, probs)))}'
 
 demo = gr.Interface(fn=is_it_a_beaver_dam, inputs=gr.Image(shape=(200, 200)), outputs=gr.Label())