Update utils/main_model.py

utils/main_model.py

@@ -8,11 +8,13 @@ class ChemicalConverter:
         self.mode = mode
         model_directory = os.path.abspath("models")
         model_path = os.path.join(model_directory, mode)
-        if not os.path.exists(model_path):
-            raise ValueError(f"Model path does not exist: {model_path}")
+        if mode == "SMILES2IUPAC":
+            model_path = "knowledgator/SMILES2IUPAC-canonical-base"
+        else:
+            model_path = "knowledgator/IUPAC2SMILES-canonical-base"
         self.model = MT5ForConditionalGeneration.from_pretrained(model_path)
-        self.smiles_tokenizer = AutoTokenizer.from_pretrained("BioMike/smiles")
-        self.iupac_tokenizer = AutoTokenizer.from_pretrained("BioMike/iupac")
+        self.smiles_tokenizer = AutoTokenizer.from_pretrained("knowledgator/SMILES-FAST-TOKENIZER")
+        self.iupac_tokenizer = AutoTokenizer.from_pretrained("knowledgator/IUPAC-FAST-TOKENIZER")
         self.smiles_max_len = 128
         self.iupac_max_len = 156