EddieXIA commited on
Commit
312f9a4
1 Parent(s): 546aff0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ access_token = "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"
12
  model_name = "bigcode/starcoderplus"
13
 
14
  # Load StarCoder model and tokenizer
15
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name, use_auth_token=access_token)
16
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=access_token)
17
 
18
  # Streamlit app
 
12
  model_name = "bigcode/starcoderplus"
13
 
14
  # Load StarCoder model and tokenizer
15
+ model = AutoModel.from_pretrained(model_name, use_auth_token=access_token)
16
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=access_token)
17
 
18
  # Streamlit app