333

app.py

@@ -12,7 +12,7 @@ def greet(img):
     a = numb % 3
     return a, numb
 
-def process_sketch(sketch):
+def process_sketch(img):
     img = np.expand_dims(img, axis=0)
     output1 = np.argmax(model.predict(img)[0])
     output2 = (np.argmax(model.predict(img)[0]) % 3)
@@ -22,9 +22,9 @@ def process_sketch(sketch):
     #demo.launch()
 
 iface = gr.Interface(
-    fn=greet,
+    fn=process_sketch,
     inputs="sketchpad",
-    outputs=["text", "text"],
+    outputs=[gr.outputs.Text(label="Среднее"), gr.outputs.Text(label="Стандартное отклонение")],
     layout="vertical",
     title="Распознавание рисунка"
 )