Update carebridge_client.py

carebridge_client.py

@@ -22,20 +22,8 @@ class CareBridgeTranslator:
         self.processor = None
         print(f"[SIMBOTI] Translator initialized. Model will load on first use.")
 
-    def _load_model(self):
-        if self.model is None:
-            print(f"[SIMBOTI] Loading model {self.model_id}...")
-            self.processor = AutoProcessor.from_pretrained(self.model_id)
-            self.model = AutoModelForImageTextToText.from_pretrained(
-                self.model_id, 
-                device_map=self.device, 
-                torch_dtype=torch.float16 if self.device == "cuda" else torch.float32
-            )
-            print("[SIMBOTI] Model loaded successfully.")
-
         # Top 10 NHS Languages Mapping (ISO 639-1)
         self.LANG_MAP = {
-
             "English": "en",
             "Polish": "pl",
             "Romanian": "ro",
@@ -49,6 +37,17 @@ class CareBridgeTranslator:
             "Italian": "it"
         }
 
+    def _load_model(self):
+        if self.model is None:
+            print(f"[SIMBOTI] Loading model {self.model_id}...")
+            self.processor = AutoProcessor.from_pretrained(self.model_id)
+            self.model = AutoModelForImageTextToText.from_pretrained(
+                self.model_id, 
+                device_map=self.device, 
+                torch_dtype=torch.float16 if self.device == "cuda" else torch.float32
+            )
+            print("[SIMBOTI] Model loaded successfully.")
+
     def translate_text(self, text, source_lang_name, target_lang_name):
         """
         Translate text ensuring patient data stays local.
@@ -184,4 +183,4 @@ class CareBridgeTranslator:
 # Simple Verification Test if run directly
 if __name__ == "__main__":
     translator = CareBridgeTranslator()
-    print("Test 1 (Text):", translator.translate_text("Where does it hurt?", "English", "Polish"))
+    print("Test 1 (Text):", translator.translate_text("Where does it hurt?", "English", "Polish"))