Update app/utils.py

app/utils.py

@@ -1,6 +1,5 @@
 import os
 from transformers import AutoModel, AutoTokenizer
-from transformers_modules.RufusRubin777.GOT_OCR2_0_CPU.modeling_GOT import GOTModel, GOTConfig
 
 class OCRModel:
     _instance = None
@@ -12,22 +11,26 @@ class OCRModel:
         return cls._instance
 
     def initialize(self):
+        # تحميل النموذج مرة واحدة وتخزينه محلياً
         model_path = os.getenv('MODEL_PATH', 'RufusRubin777/GOT-OCR2_0_CPU')
         
-        # تحميل النموذج بالطريقة الصحيحة
-        config = GOTConfig.from_pretrained(model_path)
-        self.model = GOTModel.from_pretrained(
-            model_path,
-            config=config,
-            local_files_only=False
+        self.tokenizer = AutoTokenizer.from_pretrained(
+            model_path, 
+            trust_remote_code=True,
+            local_files_only=False  # سيتم تحميل الملفات إذا لم تكن موجودة
         )
         
-        self.tokenizer = AutoTokenizer.from_pretrained(
+        self.model = AutoModel.from_pretrained(
             model_path,
-            local_files_only=False
+            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.eval()
+        self.model = self.model.eval()
+
         
     def process_image(self, image_stream):
         try: