Update app.py

app.py

@@ -2,13 +2,13 @@ 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")
@@ -39,7 +39,7 @@ def run():
 
     btn.click(one_shot, inputs=[image_in,input_text,gender], outputs=[video_out])
     demo.queue()
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)
 
 if __name__ == "__main__":
     run()