EddieXIA commited on
Commit
6424758
1 Parent(s): e81576c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -5,10 +5,12 @@ import PyPDF2
5
  import pytesseract
6
  from PIL import Image
7
 
 
 
 
8
  # Load StarCoder model and tokenizer
9
- model_name = "bigcode/starcoderplus" # Replace with the correct model name
10
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
11
- tokenizer = AutoTokenizer.from_pretrained(model_name)
12
 
13
  # Streamlit app
14
  st.title("Conversational AI Chatbot")
 
5
  import pytesseract
6
  from PIL import Image
7
 
8
+ # Your Hugging Face access token
9
+ access_token = "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"
10
+
11
  # Load StarCoder model and tokenizer
12
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name, use_auth_token=access_token)
13
+ tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=access_token)
 
14
 
15
  # Streamlit app
16
  st.title("Conversational AI Chatbot")