Update app.py

app.py

@@ -76,6 +76,11 @@ Category List : ["Dairy & Eggs", "Beverages & Snacks", "Cleaning & Hygiene", "Gr
 '''
 
 @spaces.GPU()
+num_elements = (5 * 1024 * 1024) // 4
+# Create a tensor with the calculated number of elements
+tensor = torch.randn(num_elements, dtype=torch.float32)
+# Move the tensor to the GPU
+tensor_gpu = tensor.to('cuda')
 def chunk_it(inventory_list, user_input_text):
     logger.info("Loading model and tokenizer...")
     try: