Update app/utils.py

app/utils.py

@@ -1,36 +1,33 @@
-from transformers import AutoModel, AutoTokenizer
 import os
-import torch
-from PIL import Image
+from transformers import AutoModel, AutoTokenizer
+from transformers_modules.RufusRubin777.GOT_OCR2_0_CPU.modeling_GOT import GOTModel, GOTConfig
 
 class OCRModel:
     _instance = None
-    
+
     def __new__(cls):
         if cls._instance is None:
             cls._instance = super(OCRModel, cls).__new__(cls)
             cls._instance.initialize()
         return cls._instance
-    
+
     def initialize(self):
         model_path = os.getenv('MODEL_PATH', 'RufusRubin777/GOT-OCR2_0_CPU')
         
-        self.tokenizer = AutoTokenizer.from_pretrained(
-            model_path, 
-            trust_remote_code=True,
+        # تحميل النموذج بالطريقة الصحيحة
+        config = GOTConfig.from_pretrained(model_path)
+        self.model = GOTModel.from_pretrained(
+            model_path,
+            config=config,
             local_files_only=False
         )
         
-        self.model = AutoModel.from_pretrained(
+        self.tokenizer = AutoTokenizer.from_pretrained(
             model_path,
-            trust_remote_code=True,
-            low_cpu_mem_usage=True,
-            device_map='cpu',
-            use_safetensors=True,
-            pad_token_id=self.tokenizer.eos_token_id
+            local_files_only=False
         )
         
-        self.model = self.model.eval()
+        self.model.eval()
         
     def process_image(self, image_stream):
         try: