Update app.py

app.py

@@ -9,8 +9,12 @@ import numpy as np
 print(f"torch_version: {torch.__version__}")
 
 # Define two matrices using NumPy arrays
-A = np.array([[1, 2], [3, 4]])
-B = np.array([[5, 6], [7, 8]])
+#A = np.array([[1, 2], [3, 4]])
+#B = np.array([[5, 6], [7, 8]])
+
+# Define large matrices
+A = np.random.rand(1000, 1000)  # Random 1000x1000 matrix
+B = np.random.rand(1000, 1000)
 
 # Start the timer
 start_time = time.time()
@@ -24,9 +28,14 @@ end_time = time.time()
 # Calculate and print the time taken
 print(f"Time taken for matrix multiplication with NumPy: {end_time - start_time:.6f} seconds")
 
+
 # Define two matrices
-A = torch.tensor([[1, 2], [3, 4]])
-B = torch.tensor([[5, 6], [7, 8]])
+#A = torch.tensor([[1, 2], [3, 4]])
+#B = torch.tensor([[5, 6], [7, 8]])
+
+# Define large matrices
+A = torch.rand(1000, 1000)  # Random 1000x1000 matrix
+B = torch.rand(1000, 1000)
 
 # Start the timer
 start_time = time.time()
@@ -43,8 +52,12 @@ print(f"Time taken for matrix multiplication with PyTorch: {end_time - start_tim
 @spaces.GPU
 def zeroGPU_test(text):
     # Define two matrices
-    A = torch.tensor([[1, 2], [3, 4]])
-    B = torch.tensor([[5, 6], [7, 8]])
+    #A = torch.tensor([[1, 2], [3, 4]])
+    #B = torch.tensor([[5, 6], [7, 8]])
+
+    # Define large matrices
+    A = torch.rand(1000, 1000)  # Random 1000x1000 matrix
+    B = torch.rand(1000, 1000)
     
     # Start the timer
     start_time = time.time()