update

app.py

@@ -20,9 +20,10 @@ with gr.Blocks() as demo:
         with gr.Column():
             with gr.Row():
                 a2v_input_audio = gr.Audio(sources=["upload", "microphone"], type="filepath", editable=True, label="Input audio", interactive=True)
-                with gr.Column():
-                    a2v_ref_img = gr.Image(label="Upload reference image", sources="upload")
-                    a2v_img_trans_real_botton = gr.Button("Translate to realistic style")
+                # with gr.Column():
+                #     a2v_ref_img = gr.Image(label="Upload reference image", sources="upload")
+                #     a2v_img_trans_real_botton = gr.Button("Translate to realistic style")
+                a2v_ref_img = gr.Image(label="Upload reference image", sources="upload")
                 a2v_headpose_video = gr.Video(label="Option: upload head pose reference video", sources="upload")
 
             with gr.Row():
@@ -40,9 +41,10 @@ with gr.Blocks() as demo:
     with gr.Tab("Video2video"):
          with gr.Column():
             with gr.Row():
-                with gr.Column():
-                    v2v_ref_img = gr.Image(label="Upload reference image", sources="upload")
-                    v2v_img_trans_real_botton = gr.Button("Translate to realistic style")
+                # with gr.Column():
+                #     v2v_ref_img = gr.Image(label="Upload reference image", sources="upload")
+                #     v2v_img_trans_real_botton = gr.Button("Translate to realistic style")
+                v2v_ref_img = gr.Image(label="Upload reference image", sources="upload")
                 v2v_source_video = gr.Video(label="Upload source video", sources="upload")
             
             with gr.Row():
@@ -55,7 +57,6 @@ with gr.Blocks() as demo:
             
             v2v_botton = gr.Button("Generate", variant="primary")
             v2v_output_video = gr.PlayableVideo(label="Result", interactive=False)
-            v2v_bg_restore_botton = gr.Button("Background restoration with SAM")
             
     a2v_botton.click(
         fn=audio2video,