Stanlito commited on
Commit
ee948a6
·
verified ·
1 Parent(s): 4a0156e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ from pdf2image import convert_from_path
7
  import spaces
8
 
9
  # Load the model and processor
10
- processor = NougatProcessor.from_pretrained("MohamedRashad/arabic-small-nougat")
11
- model = VisionEncoderDecoderModel.from_pretrained("MohamedRashad/arabic-small-nougat")
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
  model.to(device)
14
 
 
7
  import spaces
8
 
9
  # Load the model and processor
10
+ processor = NougatProcessor.from_pretrained("facebook/nougat-small")
11
+ model = VisionEncoderDecoderModel.from_pretrained("facebook/nougat-small")
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
  model.to(device)
14