Update app.py

app.py

@@ -56,8 +56,8 @@ def zeroGPU_test(text):
     #B = torch.tensor([[5, 6], [7, 8]])
 
     # Define large matrices
-    A = torch.rand(10000, 10000)  # Random 10000x10000 matrix
-    B = torch.rand(10000, 10000)
+    A = torch.rand(10000, 10000).to('cuda')   # Random 10000x10000 matrix
+    B = torch.rand(10000, 10000).to('cuda') 
     
     # Start the timer
     start_time = time.time()