Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,9 @@ import gradio as gr
|
|
5 |
import tempfile
|
6 |
|
7 |
# Load the OCR model and tokenizer
|
8 |
-
tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2.0',
|
|
|
|
|
9 |
model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2.0',
|
10 |
trust_remote_code=True,
|
11 |
low_cpu_mem_usage=True,
|
|
|
5 |
import tempfile
|
6 |
|
7 |
# Load the OCR model and tokenizer
|
8 |
+
tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2.0',
|
9 |
+
trust_remote_code=True,
|
10 |
+
token="<your_access_token>")
|
11 |
model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2.0',
|
12 |
trust_remote_code=True,
|
13 |
low_cpu_mem_usage=True,
|