Update app.py

app.py

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