Update load_model.py

load_model.py

@@ -14,7 +14,7 @@ def load_model(model_name):
     return model
 
 def load_tokenizer(model_name):
-    tokenizer = BertTokenizerFast.from_pretrained(model_name, use_auth_token=os.getenv('API_KEY')
+    tokenizer = BertTokenizerFast.from_pretrained(model_name, use_auth_token=os.getenv('API_KEY'))
     return tokenizer
 
 def predict(text, model, tokenizer):
@@ -23,7 +23,7 @@ def predict(text, model, tokenizer):
     return outputs
 
 def main():
-    model_name = os.getenv('Erfan11/Neuracraft')
+    model_name = os.getenv('MODEL_PATH')
     if model_name is None:
         raise ValueError("MODEL_PATH environment variable not set or is None")