Update app.py

app.py

@@ -7,7 +7,10 @@ model = from_pretrained_keras("ISYS/MyNewModel")
 
 def greet(img):
     img = np.expand_dims(img, axis=0)
-    return np.argmax(model.predict(img)[0])
+    #return np.argmax(model.predict(img)[0])
+    numb = np.argmax(model.predict(img)[0])
+    ost_numb = numb % 3
+    return ost_numb
 
 demo = gr.Interface(fn=greet, inputs="sketchpad", outputs="text")
 demo.launch()