333

app.py

@@ -12,15 +12,12 @@ def greet(img):
     a = numb % 3
     return a, numb
 
-def rasp_numb(img):
+def process_sketch(sketch):
     img = np.expand_dims(img, axis=0)
-    return np.argmax(model.predict(img)[0])
+    sketch = np.array(sketch)
+    output1 = np.argmax(model.predict(img)[0])
+    output2 = (np.argmax(model.predict(img)[0]) % 3)
+    return output1, output2
 
-outputs = [
-    gr.Textbox(label="hexadecimal color"),
-    gr.Textbox(label="hexadecimal color")
-]
-
-demo = gr.Interface(fn=greet, inputs="sketchpad", output=outputs)
-if __name__ == "__main__":
+demo = gr.Interface(fn=process_sketch, inputs="sketchpad", outputs=["text", "text"], title="Распознавание рисунка")
     demo.launch()