a

.ipynb_checkpoints/app-checkpoint.py

@@ -104,7 +104,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
 #         preds=str(outputs.logits.softmax(1).argmax(1))
 #         labels=str(data['labels'])
         
-        return img, preds, labels
+        return preds, preds, labels
     
     else:
         capture = cv2.VideoCapture(video)
@@ -136,7 +136,7 @@ demo = gr.Interface(video_identity,
                             gr.Text(),
                             gr.Text(label='Which set is this? (type train or test)'),
                             gr.Text(label='Are you ready? (type yes or no)')],
-                    outputs=[gr.Image(),
+                    outputs=[gr.Text(),
                              gr.Text(),
                              gr.Text()],
                     cache_examples=True)

app.py

@@ -104,7 +104,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
 #         preds=str(outputs.logits.softmax(1).argmax(1))
 #         labels=str(data['labels'])
         
-        return img, preds, labels
+        return preds, preds, labels
     
     else:
         capture = cv2.VideoCapture(video)
@@ -136,7 +136,7 @@ demo = gr.Interface(video_identity,
                             gr.Text(),
                             gr.Text(label='Which set is this? (type train or test)'),
                             gr.Text(label='Are you ready? (type yes or no)')],
-                    outputs=[gr.Image(),
+                    outputs=[gr.Text(),
                              gr.Text(),
                              gr.Text()],
                     cache_examples=True)