Update old_version.py

old_version.py

@@ -26,11 +26,12 @@ all_stopwords = stopwords.words('english') # Making sure to only use English sto
 extra_stopwords = ['ii', 'iii'] # Can add extra stopwords to be removed from dataset/input abstracts
 all_stopwords.extend(extra_stopwords)
 
+modelpath = os.environ.get("MODEL_PATH")
 ########### GET CLAIMED TRAINED MODEL ###########
-tokenizer = LlamaTokenizer.from_pretrained("samwit/koala-7b")
+tokenizer = LlamaTokenizer.from_pretrained(modelpath)
 
 model = LlamaForCausalLM.from_pretrained(
-    "samwit/koala-7b",
+    modelpath,
     load_in_8bit=True,
     device_map='auto',
 )