Update app.py

app.py

@@ -1,32 +1,22 @@
 import gradio as gr
-import spaces
 import torch
 
-spaces.zero.torch.unpatch()
 
-print('cuda avaliable: ' + str(torch.cuda.is_available()))
-torch.set_default_device('cuda')
+def greet(n):
+    print('cuda avaliable: ' + str(torch.cuda.is_available()))
 
-zero = torch.Tensor([0]).cuda()
+    torch.set_default_device('cuda')
 
-one = torch.ones(2, 2, device='cuda')
+    zero = torch.Tensor([0]).cuda()
 
-two = torch.matmul(one, one).cuda()
+    one = torch.ones(2, 2, device='cuda')
 
+    two = torch.matmul(one, one).cuda()
 
-def print_device():
     print('zero device: ' + str(zero.device))
     print('one device: ' + str(one.device))
     print('two device: ' + str(two.device))
 
-
-print_device()
-
-
-@spaces.GPU
-def greet(n):
-    print('greeting')
-    print_device()
     return f"Hello {zero + n} Tensor"