EddieXIA commited on
Commit
00ca433
1 Parent(s): d53739c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  # Import necessary libraries
2
- import streamlit as st
3
  from transformers import AutoModel, AutoTokenizer
 
4
  import PyPDF2
5
  import pytesseract
6
  from PIL import Image
@@ -11,7 +11,7 @@ access_token = "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"
11
  # Correct model name or path
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
 
 
1
  # Import necessary libraries
 
2
  from transformers import AutoModel, AutoTokenizer
3
+ import streamlit as st
4
  import PyPDF2
5
  import pytesseract
6
  from PIL import Image
 
11
  # Correct model name or path
12
  model_name = "bigcode/starcoderplus"
13
 
14
+ # Load 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