Update app.py

app.py

@@ -22,7 +22,7 @@ class AtlasOCR:
             logger.info(f"Loading model: {model_name}")
             self.model, self.processor = FastVisionModel.from_pretrained(
                 model_name,
-                device_map="auto",
+                device_map="cuda",
                 load_in_4bit=True,
                 use_gradient_checkpointing="unsloth",
                 token=os.environ["HF_API_KEY"]