Update app.py

app.py

@@ -38,7 +38,7 @@ def predict(image):
   outputs = model.generate(
       pixel_values.to(device),
       min_length=1,
-      max_new_tokens=1000,
+      max_new_tokens=1500,
       bad_words_ids=[[processor.tokenizer.unk_token_id]],
   )
 
@@ -62,7 +62,7 @@ def inference(pdf_file, pdf_link):
   images = rasterize_paper(file_name, return_pil=True)
   sequence = ""
   # infer for every page and concat
-  for image in images[0:2]:
+  for image in images:
     sequence += predict(image)