Update app.py

app.py

@@ -2,14 +2,19 @@ import gradio as gr
 import spaces
 import torch
 
-zero = torch.Tensor([0]).cuda()
-print(zero.device) # <-- 'cpu' 🤔
+# zero = torch.Tensor([0]).cuda()
+# print(zero.device) # <-- 'cpu' 🤔
 
-@spaces.GPU
-def greet(n):
-    print(zero.device) # <-- 'cuda:0' 🤗
-    return f"Hello {zero + n} Tensor"
+# @spaces.GPU
+# 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():
   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/>")
@@ -32,7 +37,7 @@ def run():
 
     #         """)
 
-    # btn.click(one_shot, inputs=[image_in,input_text,gender], outputs=[video_out])
+    btn.click(one_shot, inputs=[image_in,input_text,gender], outputs=[video_out])
     demo.queue()
 
 if __name__ == "__main__":