a

.ipynb_checkpoints/app-checkpoint.py

@@ -4,21 +4,13 @@ import cv2
 from encoded_video import EncodedVideo, write_video
 import torch
 import numpy as np
-import gc
 
 def video_identity(video):
 
-    # gc.collect()
-    # output_video_path = "./output.mp4"
-    capture = cv2.VideoCapture(video)
-#     fourcc = cv2.VideoWriter_fourcc(*"I420")  # encoder
 
-#     out = cv2.VideoWriter(output_video_path, fourcc, 30.0, (int(cap.get(3)), int(cap.get(4))))
-    
-    
+    capture = cv2.VideoCapture(video, text)
      
     frameNr = 0
-    # capture = cv2.VideoCapture('output.mp4')
     while (True):
     
         success, frame = capture.read()
@@ -35,7 +27,8 @@ def video_identity(video):
     
     return img
 demo = gr.Interface(video_identity, 
-                    gr.Video(source='upload'), 
-                    gr.Image(), 
+                    inputs=[gr.Video(source='upload'),
+                            gr.Text()],
+                    outputs=gr.Image(), 
                     cache_examples=True)
 demo.launch(debug=True)

app.py

@@ -4,21 +4,13 @@ import cv2
 from encoded_video import EncodedVideo, write_video
 import torch
 import numpy as np
-import gc
 
 def video_identity(video):
 
-    # gc.collect()
-    # output_video_path = "./output.mp4"
-    capture = cv2.VideoCapture(video)
-#     fourcc = cv2.VideoWriter_fourcc(*"I420")  # encoder
 
-#     out = cv2.VideoWriter(output_video_path, fourcc, 30.0, (int(cap.get(3)), int(cap.get(4))))
-    
-    
+    capture = cv2.VideoCapture(video, text)
      
     frameNr = 0
-    # capture = cv2.VideoCapture('output.mp4')
     while (True):
     
         success, frame = capture.read()
@@ -35,7 +27,8 @@ def video_identity(video):
     
     return img
 demo = gr.Interface(video_identity, 
-                    gr.Video(source='upload'), 
-                    gr.Image(), 
+                    inputs=[gr.Video(source='upload'),
+                            gr.Text()],
+                    outputs=gr.Image(), 
                     cache_examples=True)
 demo.launch(debug=True)