Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 
 
 ROOP_DEFAULT_PATH = "media/roop_default.png"
-ROOP_OUTPUT_VIDEO_PATH = "media/roop_output.mp4"
+ROOP_OUTPUT_PATH = "media/roop_output.png"
 
 
 def _handle_roop_solve(
@@ -20,7 +20,7 @@ def _handle_roop_solve(
     if not os.path.exists("media"):
         os.mkdir("media")
     start_time = time.time()
-    command = f"python run.py -t {video_path} -s {img_path} -o {ROOP_OUTPUT_VIDEO_PATH} --execution-threads 1024"
+    command = f"python run.py -t {video_path} -s {img_path} -o {ROOP_OUTPUT_PATH} --execution-threads 64 --execution-provider gpu"
     os.system(command)
     solved_time = time.time() - start_time
     message = "Successfully solve the GED problem, using time ({:.3f}s).".format(solved_time)
@@ -71,10 +71,15 @@ with gr.Blocks() as ged_page:
     with gr.Row(variant="panel"):
         with gr.Column(scale=2):
             with gr.Row():
-                upload_video = gr.Video(
-                    label="Upload .mp4 Vide0",
-                    format="mp4",
+                upload_video = gr.Image(
+                    label="Upload .png or .jpg File",
+                    type="filepath",
+                    min_width=40,
                 )
+                #gr.Video(
+                #    label="Upload .mp4 Vide0",
+                #    format="mp4",
+                #)
                 upload_img = gr.Image(
                     label="Upload .png or .jpg File",
                     type="filepath",