1

app.py

@@ -11,5 +11,9 @@ def greet(img):
     numb = np.argmax(model.predict(img)[0])
     return numb % 3
 
-demo = gr.Interface(fn=greet, inputs="sketchpad", outputs="text")
+def zalupa(img):
+    img = np.expand_dims(img, axis=0)
+    return np.argmax(model.predict(img)[0])
+
+demo = gr.Interface(fn=greet, inputs="sketchpad", outputs="text", outputs=zalupa)
 demo.launch()