Update app/utils.py

app/utils.py

@@ -18,9 +18,9 @@ class OCRModel:
             cls._instance = super(OCRModel, cls).__new__(cls)
             cls._instance.initialize()
         return cls._instance
+    
     def initialize(self):
         try:
-
             logger.info("Initializing OCR model...")
             # Try loading with use_fast=False if the fast tokenizer fails
             try:
@@ -41,12 +41,11 @@ class OCRModel:
                 'stepfun-ai/GOT-OCR-2.0-hf',
                 trust_remote_code=True,
                 low_cpu_mem_usage=True,
-                device_map='auto',  # Let transformers decide the best device
+                device_map='auto',
                 use_safetensors=True,
                 pad_token_id=self.tokenizer.eos_token_id
             )
             
-            # Let the model decide device placement
             self.device = "cuda" if torch.cuda.is_available() else "cpu"
             self.model = self.model.eval().to(self.device)