Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
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 |
|