Remove cuda check that might causes error

ocr_engine.py

@@ -1,6 +1,5 @@
 # Load model directly
 import os
-import torch
 from transformers import AutoModel, AutoTokenizer
 from PIL import Image
 import uuid
@@ -18,10 +17,8 @@ class OCRModel:
 
     def __init__(self):
         self.tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
-        self.model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cuda' if torch.cuda.is_available() else "cpu", use_safetensors=True, pad_token_id=self.tokenizer.eos_token_id)
+        self.model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cpu', use_safetensors=True, pad_token_id=self.tokenizer.eos_token_id)
         self.model = self.model.eval()
-        if torch.cuda.is_available():
-            self.model = self.model.cuda()
 
     def chat(self, image: Image.Image) -> str:
         unique_id = str(uuid.uuid4())
@@ -31,8 +28,7 @@ class OCRModel:
         image.save(image_path)
         
         res = self.model.chat(self.tokenizer, image_path, ocr_type='ocr')
-        with open(result_path, 'w') as f:
-            f.write(res)
+        os.remove(image_path) # delete file create to avoid memory issue and data shared online
         return res
 
 ocr_model = OCRModel()