Update app.py

app.py

@@ -5,14 +5,16 @@ import tensorflow
 
 model = from_pretrained_keras("merve/riddikulus")
 
-labels = {0:"ravenclaw", 1:"gryffindor", 2:"slytherin",3:"ravenclaw",
-          4:"ravenclaw", 5:"gryffindor", 6: "hufflepuff", 7:"ravenclaw",
-          8:"ravenclaw", 9:"slytherin"}
+labels = {0:"Ravenclaw 🦅💙 ", 1:"Gryffindor 🦁", 2:"Ravenclaw 🦅💙",3:"Slytherin🐍💚",
+          4:"Hufflepuff 🦡💛", 5:"Gryffindor 🦁 ", 6: "Hufflepuff 🦡💛", 7:"Slytherin🐍💚",
+          8:"Ravenclaw 🦅💙", 9:"Death eater detected!"}
           
 canvas = gr.inputs.Image(source="canvas", shape=(28,28))          
 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! 💀"
   cls = labels[cls]
   return f"Welcome to {cls}"