Update app.py

app.py

@@ -14,9 +14,10 @@ text = gr.outputs.Textbox()
 def infer(image):
   cls = np.argmax(model.predict(np.expand_dims(image, axis = 0)[:,:,:,1]))
   if cls == 9:
-    return "Death eater detected! 💀"
+    output = "Death eater detected! 💀"
   else:
     cls = labels[cls]
-    return f"Welcome to {cls}"
+    output = f"Welcome to {cls}"
+  return output
   
 gr.Interface(infer, inputs=[canvas], outputs=[text], title="Welcome to Hogwarts Sorting Hat!", description="Draw something and let the sorting hat sort you! 🎩 ").launch()