Update app.py

app.py

@@ -1,7 +1,22 @@
+import gradio as gr
+import spaces
+import torch
+
+
+#@spaces.GPU
+#def greet(n):
+#    print(zero.device) # <-- 'cuda:0' 🤗
+#    return f"Hello {zero + n} Tensor"
+
+#demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
+#demo.launch()
+
 import os
 import subprocess
 import urllib.request
 import torch
+zero = torch.Tensor([0]).cuda()
+print(zero.device) # <-- 'cpu' 🤔
 print("Torch version:", torch.__version__)
 if torch.cuda.is_available():
     cap = torch.cuda.get_device_capability(0)