a

.ipynb_checkpoints/app-checkpoint.py

@@ -6,26 +6,10 @@ import cv2
 
 def video_identity(video):
     
-    capture=cv2.VideoCapture(video.name)
-    frameNr = 0
- 
-    while (True):
 
-        success, frame = capture.read()
-
-        if success:
-            cv2.imwrite(f'train/frame_{frameNr}.jpg', frame)
-
-        else:
-            break
-
-        frameNr = frameNr+10
-
-    capture.release()
-    
-    return str(frameNr)
+    return video
 demo = gr.Interface(video_identity, 
                     gr.Video(), 
-                    'text', 
+                    gr.Video(), 
                     cache_examples=True)
 demo.launch(debug=True)

app.py

@@ -6,26 +6,10 @@ import cv2
 
 def video_identity(video):
     
-    capture=cv2.VideoCapture(video.name)
-    frameNr = 0
- 
-    while (True):
 
-        success, frame = capture.read()
-
-        if success:
-            cv2.imwrite(f'train/frame_{frameNr}.jpg', frame)
-
-        else:
-            break
-
-        frameNr = frameNr+10
-
-    capture.release()
-    
-    return str(frameNr)
+    return video
 demo = gr.Interface(video_identity, 
                     gr.Video(), 
-                    'text', 
+                    gr.Video(), 
                     cache_examples=True)
 demo.launch(debug=True)