Update app.py

app.py

@@ -6,14 +6,14 @@ import tensorflow
 model = from_pretrained_keras("merve/riddikulus")
 
 labels = {0:"Ravenclaw 🦅💙 ", 1:"Gryffindor 🦁", 2:"Ravenclaw 🦅💙",3:"Slytherin🐍💚",
-          4:"Hufflepuff 🦡💛", 9:"Death eater detected! ", 6: "Hufflepuff 🦡💛", 7:"Slytherin🐍💚",
-          8:"Ravenclaw 🦅💙", 5:"Gryffindor 🦁"}
+          4:"Hufflepuff 🦡💛", 5:"Death eater detected! ", 6: "Hufflepuff 🦡💛", 7:"Slytherin🐍💚",
+          8:"Ravenclaw 🦅💙", 9:"Gryffindor 🦁"}
           
 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:
+  if cls == 5:
     output = "Death eater detected! 💀"
   else:
     cls = labels[cls]