Update app.py

app.py

@@ -38,13 +38,12 @@ def predict(image):
   outputs = model.generate(
       pixel_values.to(device),
       min_length=1,
-      max_new_tokens=30,
       bad_words_ids=[[processor.tokenizer.unk_token_id]],
   )
 
-  sequence = processor.batch_decode(outputs, skip_special_tokens=True)[0]
-  sequence = processor.post_process_generation(sequence, fix_markdown=False)
-  return sequence
+  page_sequence = processor.batch_decode(outputs, skip_special_tokens=True)[0]
+  page_sequence = processor.post_process_generation(sequence, fix_markdown=False)
+  return page_sequence