Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,8 @@ class AtlasOCR:
|
|
| 24 |
model_name,
|
| 25 |
device_map="auto",
|
| 26 |
load_in_4bit=True,
|
| 27 |
-
use_gradient_checkpointing="unsloth"
|
|
|
|
| 28 |
)
|
| 29 |
self.max_tokens = max_tokens
|
| 30 |
self.prompt = ""
|
|
|
|
| 24 |
model_name,
|
| 25 |
device_map="auto",
|
| 26 |
load_in_4bit=True,
|
| 27 |
+
use_gradient_checkpointing="unsloth",
|
| 28 |
+
token=os.environ["HF_API_KEY"]
|
| 29 |
)
|
| 30 |
self.max_tokens = max_tokens
|
| 31 |
self.prompt = ""
|