update gradio

app.py

@@ -50,12 +50,12 @@ def sadtalker_demo():
         <br/><b>Alternatively, try our GitHub <a href=https://github.com/Winfredy/SadTalker> code </a> on your own GPU. </b> <a style='display:inline-block' href="https://github.com/Winfredy/SadTalker"><img src="https://img.shields.io/github/stars/Winfredy/SadTalker?style=social"/></a> \
         """)
         
-        with gr.Row().style(equal_height=False):
+        with gr.Row():
             with gr.Column(variant='panel'):
                 with gr.Tabs(elem_id="sadtalker_source_image"):
                     with gr.TabItem('Source image'):
                         with gr.Row():
-                            source_image = gr.Image(label="Source image", source="upload", type="filepath", elem_id="img2img_image").style(width=512)
+                            source_image = gr.Image(label="Source image", elem_id="img2img_image")
 
 
                 with gr.Tabs(elem_id="sadtalker_driven_audio"):
@@ -63,8 +63,8 @@ def sadtalker_demo():
                         gr.Markdown("Possible driving combinations: <br> 1. Audio only 2. Audio/IDLE Mode + Ref Video(pose, blink, pose+blink) 3. IDLE Mode only 4. Ref Video only (all) ")
 
                         with gr.Row():
-                            driven_audio = gr.Audio(label="Input audio", source="upload", type="filepath")
-                            driven_audio_no = gr.Audio(label="Use IDLE mode, no audio is required", source="upload", type="filepath", visible=False)
+                            driven_audio = gr.Audio(label="Input audio")
+                            driven_audio_no = gr.Audio(label="Use IDLE mode, no audio is required", visible=False)
 
                             with gr.Column():
                                 use_idle_mode = gr.Checkbox(label="Use Idle Animation")
@@ -72,7 +72,7 @@ def sadtalker_demo():
                                 use_idle_mode.change(toggle_audio_file, inputs=use_idle_mode, outputs=[driven_audio, driven_audio_no]) # todo
 
                         with gr.Row():
-                            ref_video = gr.Video(label="Reference Video", source="upload", type="filepath", elem_id="vidref").style(width=512)
+                            ref_video = gr.Video(label="Reference Video", elem_id="vidref")
 
                             with gr.Column():
                                 use_ref_video = gr.Checkbox(label="Use Reference Video")
@@ -108,7 +108,7 @@ def sadtalker_demo():
                             submit = gr.Button('Generate', elem_id="sadtalker_generate", variant='primary')
                             
                 with gr.Tabs(elem_id="sadtalker_genearted"):
-                        gen_video = gr.Video(label="Generated video", format="mp4").style(width=256)
+                        gen_video = gr.Video(label="Generated video", format="mp4")
 
         
 
@@ -218,6 +218,6 @@ if __name__ == "__main__":
 
     demo = sadtalker_demo()
     demo.queue(max_size=10, api_open=True)
-    demo.launch(debug=True)
+    demo.launch(debug=True, enable_queue=False)
 
 

requirements.txt

@@ -1,8 +1,8 @@
-torch==1.13.1
-torchvision==0.14.1
-torchaudio==0.13.1
+torch==2.0.1
+torchvision==0.15.2
+torchaudio==2.0.2
 numpy==1.23.5
-face_alignment==1.3.0
+face_alignment==1.4.1
 imageio==2.19.3
 imageio-ffmpeg==0.4.7
 librosa==0.8.0