Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import spaces
 import torch
 
+torch.set_default_device('cuda')
+
 zero = torch.Tensor([0]).cuda()
 
 one = torch.ones(2, 2, device='cuda')
@@ -11,6 +13,7 @@ two = torch.matmul(one, one).cuda()
 
 def print_device():
     print('cuda avaliable: ' + str(torch.cuda.is_available()))
+    print('default device: ' + str(torch.get_default_device()))
     print('zero device: ' + str(zero.device))
     print('one device: ' + str(one.device))
     print('two device: ' + str(two.device))