ahmed-masry commited on
Commit
773d067
·
verified ·
1 Parent(s): 2ae9a9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ torch.hub.download_url_to_file('https://raw.githubusercontent.com/vis-nlp/ChartQ
9
 
10
 
11
  model_name = "ahmed-masry/unichart-base-960"
12
- model = VisionEncoderDecoderModel.from_pretrained(model_name, use_auth_token=os.environ['temp_access_token'])
13
- processor = DonutProcessor.from_pretrained(model_name, use_auth_token=os.environ['temp_access_token'])
14
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15
  model.to(device)
16
 
 
9
 
10
 
11
  model_name = "ahmed-masry/unichart-base-960"
12
+ model = VisionEncoderDecoderModel.from_pretrained(model_name)
13
+ processor = DonutProcessor.from_pretrained(model_name)
14
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15
  model.to(device)
16