Update app.py

app.py

@@ -51,6 +51,7 @@ install_cuda_toolkit()
 
     
 print(f"GPU: {torch.cuda.is_available()}")
+a = torch.tensor([0]).cuda()
 if torch.cuda.is_available() and torch.cuda.device_count() >= 2:
     device0 = torch.device('cuda:0')
     device1 = torch.device('cuda:0')