Update app.py

app.py

@@ -7,8 +7,8 @@ from nougat.dataset.rasterize import rasterize_paper
 from transformers import NougatProcessor, VisionEncoderDecoderModel
 import torch
 
-processor = NougatProcessor.from_pretrained("nielsr/nougat")
-model = VisionEncoderDecoderModel.from_pretrained("nielsr/nougat")
+processor = NougatProcessor.from_pretrained("facebook/nougat-small")
+model = VisionEncoderDecoderModel.from_pretrained("facebook/nougat-small")
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)