Update app.py

app.py

@@ -31,6 +31,13 @@ def upload2aws(img_array):
     s3.put_object(Bucket='predict-packages', Key=f'images_webapp_counters/{unique_name}.jpg', Body=buffer)
     return None
 
+def vidupload2aws(vid_path): 
+    vid_name = os.path.basename(vid_path)
+    _, ext = os.path.splitext(vid_name)
+    unique_name = str(uuid.uuid4())
+    s3.upload_file(vid_path, 'predict-packages', f'images_webapp_counters/videos/{unique_name}{ext}')
+    return None
+
 def send2api(input_img, api_url):
     buf = io.BytesIO()
     plt.imsave(buf, input_img, format='jpg')
@@ -138,8 +145,8 @@ def testing_directcounter(input_img):
     return img, text
 
 def video_identity(video):
-    
-    return video, print(type(video), video)
+    vidupload2aws(video)
+    return video
 
 with gr.Blocks() as demo:
     gr.Markdown("Submit an image with insects in a trap")