a

.ipynb_checkpoints/app-checkpoint.py

@@ -7,12 +7,15 @@ import numpy as np
 
 def video_identity(video,user_name,class_name,trainortest,ready):
     if ready=='yes':
-        return user_name
+        
+        img=cv2.imread('train/book/frame_0.jpg')
+        img=cv2.imread('train/wallet/frame_0.jpg')
+        return img, class_name
     else:
         capture = cv2.VideoCapture(video)
    
-        user_d=str(trainortest)+str(user_name)
-        class_d=str(trainortest)+str(user_name)+'/'+str(class_name)
+        user_d=str(trainortest)+'/'+str(user_name)
+        class_d=str(trainortest)+'/'+str(user_name)+'/'+str(class_name)
         if not os.path.exists(user_d):
             os.makedirs(user_d) 
         if not os.path.exists(class_d):
@@ -32,14 +35,15 @@ def video_identity(video,user_name,class_name,trainortest,ready):
 
         img=cv2.imread(class_d+'/frame_0.jpg')
 
-        return img, class_d
+        return img, img, class_d
 demo = gr.Interface(video_identity, 
                     inputs=[gr.Video(source='upload'),
                             gr.Text(),
                             gr.Text(),
-                            gr.Text(label='type train or test'),
-                            gr.Text(label='type yes or no')],
+                            gr.Text(label='Which set is this? (type train or test)'),
+                            gr.Text(label='Are you ready? (type yes or no)')],
                     outputs=[gr.Image(),
+                             gr.Image(),
                              gr.Text()],
                     cache_examples=True)
 demo.launch(debug=True)

app.py

@@ -7,12 +7,15 @@ import numpy as np
 
 def video_identity(video,user_name,class_name,trainortest,ready):
     if ready=='yes':
-        return user_name
+        
+        img=cv2.imread('train/book/frame_0.jpg')
+        img=cv2.imread('train/wallet/frame_0.jpg')
+        return img, class_name
     else:
         capture = cv2.VideoCapture(video)
    
-        user_d=str(trainortest)+str(user_name)
-        class_d=str(trainortest)+str(user_name)+'/'+str(class_name)
+        user_d=str(trainortest)+'/'+str(user_name)
+        class_d=str(trainortest)+'/'+str(user_name)+'/'+str(class_name)
         if not os.path.exists(user_d):
             os.makedirs(user_d) 
         if not os.path.exists(class_d):
@@ -32,14 +35,15 @@ def video_identity(video,user_name,class_name,trainortest,ready):
 
         img=cv2.imread(class_d+'/frame_0.jpg')
 
-        return img, class_d
+        return img, img, class_d
 demo = gr.Interface(video_identity, 
                     inputs=[gr.Video(source='upload'),
                             gr.Text(),
                             gr.Text(),
-                            gr.Text(label='type train or test'),
-                            gr.Text(label='type yes or no')],
+                            gr.Text(label='Which set is this? (type train or test)'),
+                            gr.Text(label='Are you ready? (type yes or no)')],
                     outputs=[gr.Image(),
+                             gr.Image(),
                              gr.Text()],
                     cache_examples=True)
 demo.launch(debug=True)