Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import spaces
 import torch
+import subprocess
 
 zero = torch.Tensor([0]).cuda()
 print(zero.device) # <-- 'cpu' 🤔
@@ -10,11 +11,19 @@ def greet(n):
     print(zero.device) # <-- 'cuda:0' 🤗
     return f"Hello {zero + n} Tensor"
 
-def one_shot(image_in,input_text,gender):
-    print("Hello")
-    return "htllo123"
+def run_infrence(input_video,input_audio):
+    audio = "sample_data/sir.mp3"
+    video = "sample_data/spark_input.mp4"
+    command = f'python3 inference.py --checkpoint_path checkpoints/wav2lip_gan.pth --face "{video}" --audio "{audio}"'
+    print("running ")
+    # Execute the command
+    process = subprocess.Popen(command, stdout=subprocess.PIPE, shell=True)
+
+    # Get the output
+    output, error = process.communicate()
+
+    return output
 
-    
 def run():
   with gr.Blocks(css=".gradio-container {background-color: lightgray} #radio_div {background-color: #FFD8B4; font-size: 40px;}") as demo:
     gr.Markdown("<h1 style='text-align: center;'>"+ "One Shot Talking Face from Text" + "</h1><br/><br/>")
@@ -22,24 +31,16 @@ def run():
       # with gr.Box():
         with gr.Row():
         # with gr.Row().style(equal_height=True):
-          image_in = gr.Image(show_label=True, type="filepath",label="Input Image")
-          input_text = gr.Textbox(show_label=True,label="Input Text")
-          gender = gr.Radio(["Female","Male"],value="Female",label="Gender")
-          video_out = gr.Video(show_label=True,label="Output")
+            input_video = gr.Video(label="Input Video")
+            input_audio = gr.Audio(label="Input Audio")
+            video_out = gr.Video(show_label=True,label="Output")
         with gr.Row():
-        # with gr.Row().style(equal_height=True):
-          btn = gr.Button("Generate")   
-    # gr.Markdown(
-    #         """
-    #         <p style='text-align: center;'>Feel free to give us your thoughts on this demo and please contact us at 
-    #                 <a href="mailto:[email protected]" target="_blank">[email protected]</a> 
-    #                 <p style='text-align: center;'>Developed by: <a href="https://www.pragnakalp.com" target="_blank">Pragnakalp Techlabs</a></p>
+            btn = gr.Button("Generate")   
 
-    #         """)
-
-    btn.click(one_shot, inputs=[image_in,input_text,gender], outputs=[video_out])
+    btn.click(run_infrence,inputs=[input_video,input_audio], outputs=[video_out])
     demo.queue()
     demo.launch(server_name="0.0.0.0", server_port=7860)
 
+
 if __name__ == "__main__":
-    run()
+    run()