zhengr commited on
Commit
9ccaae0
·
verified ·
1 Parent(s): 0af0909

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ from datetime import date, datetime, timedelta
16
  access_token = os.environ["TOKEN"]
17
 
18
  # load model
19
- model = "meta-llama/Llama-2-7b-chat-hf"
20
  peft_model = "FinGPT/fingpt-forecaster_sz50_llama2-7B_lora"
21
 
22
  tokenizer = AutoTokenizer.from_pretrained(model, token = access_token, trust_remote_code=True)
 
16
  access_token = os.environ["TOKEN"]
17
 
18
  # load model
19
+ model = "NousResearch/Llama-2-7b-chat-hf"
20
  peft_model = "FinGPT/fingpt-forecaster_sz50_llama2-7B_lora"
21
 
22
  tokenizer = AutoTokenizer.from_pretrained(model, token = access_token, trust_remote_code=True)