abdeljalilELmajjodi commited on
Commit
524600b
·
verified ·
1 Parent(s): 4035386

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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 = ""