Update app.py

app.py

@@ -28,9 +28,10 @@ all_stopwords.extend(extra_stopwords)
 
 ########### GET CLAIMED TRAINED MODEL ###########
 auth_token = os.environ.get("AUTH_TOKEN_SECRET")
-tokenizer = LlamaTokenizer.from_pretrained("Claimed/capybara", use_auth_token=auth_token)
+model_path = os.environ.get("MODEL_PATH")
+tokenizer = LlamaTokenizer.from_pretrained(model_path, use_auth_token=auth_token)
 model = LlamaForCausalLM.from_pretrained(
-     "Claimed/capybara", use_auth_token=auth_token, 
+     model_path, use_auth_token=auth_token, 
     load_in_8bit=True,
     device_map="auto") #low_cpu_mem_usage=True)