Upload app.py

app.py

@@ -29,13 +29,13 @@ def img2label(left, right):
     pred = pred.cpu().numpy().tolist()
 
     if len(pred) == 0 or (len(pred) == 1 and pred[0] == 0):
-        return LABELS[0]
+        return res_dict, LABELS[0]
     res = ''
     for i in pred:
         if i == 0:
             continue
         res += ', ' + LABELS[i]
-    return '目前的身体状态：' + res[2:]
+    return res_dict, '目前的身体状态：' + res[2:]
 
 
 if __name__ == '__main__':
@@ -71,6 +71,6 @@ if __name__ == '__main__':
     # r = img2label(left_img_dir, right_img_dir)
     demo = gr.Interface(fn=img2label,
                         inputs=[gr.inputs.Image(), gr.inputs.Image()],
-                        outputs="text",
+                        outputs=["label", "text"],
                         examples=examples, title=title, description=description)
     demo.launch()