Update utils/main_model.py

utils/main_model.py

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