Update app.py

app.py

@@ -39,7 +39,7 @@ def run_prediction(sample, model, processor, mode):
             do_sample=True,
             top_p=0.92, 
             top_k=5,
-            no_repeat_ngram_size=10,
+            no_repeat_ngram_size=15,
             num_beams=3,
             output_attentions=False,
             output_hidden_states=False,
@@ -83,7 +83,7 @@ else:
 st.image(image, caption='Your target document')
 
 with st.spinner(f'Processing the document ...'):
-        pre_trained_model = "unstructuredio/chipper-fast-fine-tuning"
+        pre_trained_model = "unstructuredio/chipper-v3-preview"
         processor = DonutProcessor.from_pretrained(pre_trained_model, token=os.environ['HF_TOKEN'])
         
         device = "cuda" if torch.cuda.is_available() else "cpu"