333

app.py

@@ -8,17 +8,18 @@ model = from_pretrained_keras("ISYS/MyNewModel")
 def greet(img):
     img = np.expand_dims(img, axis=0)
     #return np.argmax(model.predict(img)[0])
-    numb = np.argmax(model.predict(img)[0])
-    return numb % 3
+    numb = np.argmax(model.predict(img)[0]
+    a = numb % 3
+    return a, numb
 
 def rasp_numb(img):
     img = np.expand_dims(img, axis=0)
     return np.argmax(model.predict(img)[0])
 
+outputs = [
+    gr.Textbox(label="hexadecimal color"),
+    gr.Textbox(label="hexadecimal color")
+]
 
-output1 = gr.outputs.Text((np.argmax(model.predict(img)[0])% 3) )
-
-output2 = gr.outputs.Text(np.argmax(model.predict(img)[0]))
-
-demo = gr.Interface(fn=greet, inputs="sketchpad", outputs=[output1, output2])
+demo = gr.Interface(fn=greet, inputs="sketchpad", output=outputs)
 demo.launch()