Update app.py

app.py

@@ -34,7 +34,7 @@ def find_result_image(path):
     return None
 
 # --- 2. Main Processing Function (UPDATED for multi-bbox drawing) ---
-@spaces.GPU
+# @spaces.GPU
 def process_ocr_task(image, model_size, task_type, ref_text):
     """
     Processes an image with DeepSeek-OCR for all supported tasks.
@@ -44,8 +44,8 @@ def process_ocr_task(image, model_size, task_type, ref_text):
         return "Please upload an image first.", None
 
     print("🚀 Moving model to GPU...")
-    model_gpu = model.cuda().to(torch.bfloat16)
-    print("✅ Model is on GPU.")
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    model = model.to(device)
 
     with tempfile.TemporaryDirectory() as output_path:
         # Build the prompt... (same as before)