Update app.py

app.py

@@ -94,7 +94,7 @@ def get_math_response(image_description, user_question):
         text=query,
         padding=True,
         return_tensors="pt"
-    ).to("cuda")  # Use CPU if GPU is unavailable
+    ).to("cpu")  # Use CPU if GPU is unavailable
 
     # Generate the math reasoning response
     output_ids = math_model.generate(