Update app.py

app.py

@@ -1,11 +1,46 @@
 import gradio as gr
 import spaces
 import torch
+import time
 
- zero = torch.Tensor([0]).cuda()
-print(zero.device) # <-- 'cpu' 🤔
+import numpy as np
 
- @spaces.GPU
+
+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]])
+
+# Start the timer
+start_time = time.time()
+
+# Perform matrix multiplication
+result = np.dot(A, B)
+
+# End the timer
+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]])
+
+# Start the timer
+start_time = time.time()
+
+# Perform matrix multiplication
+result = torch.matmul(A, B)
+
+# End the timer
+end_time = time.time()
+
+# Calculate and print the time taken
+print(f"Time taken for matrix multiplication with PyTorch: {end_time - start_time:.6f} seconds")
+
+@spaces.GPU
 def greet(n):
     print(zero.device) # <-- 'cuda:0' 🤗
     return f"Hello {zero + n} Tensor"