thepolymerguy commited on
Commit
b81f1ae
·
1 Parent(s): 2dd90fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -28,7 +28,8 @@ all_stopwords.extend(extra_stopwords)
28
 
29
  ########### GET CLAIMED TRAINED MODEL ###########
30
  auth_token = os.environ.get("AUTH_TOKEN_SECRET")
31
- model_path = os.environ.get("MODEL_PATH")
 
32
  tokenizer = LlamaTokenizer.from_pretrained(model_path, use_auth_token=auth_token)
33
  model = LlamaForCausalLM.from_pretrained(
34
  model_path, use_auth_token=auth_token,
 
28
 
29
  ########### GET CLAIMED TRAINED MODEL ###########
30
  auth_token = os.environ.get("AUTH_TOKEN_SECRET")
31
+ #model_path = os.environ.get("MODEL_PATH")
32
+ model_path = 'eachadea/vicuna-13b'
33
  tokenizer = LlamaTokenizer.from_pretrained(model_path, use_auth_token=auth_token)
34
  model = LlamaForCausalLM.from_pretrained(
35
  model_path, use_auth_token=auth_token,