Update app.py

app.py

@@ -8,7 +8,7 @@ print('zero device: ' + str(zero.device))
 one = torch.ones(2, 2, device='cuda')
 print('one device: ' + str(one.device))
 
-two = torch.matmul(one, one, device='cuda')
+two = torch.matmul(one, one).cuda()
 print('two device: ' + str(two.device))
 
 @spaces.GPU