nougat-transformers

Runtime error

merve HF Staff commited on Sep 20, 2023

Commit

461f4fc

1 Parent(s): ec7e02b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,6 +38,7 @@ def predict(image):
   outputs = model.generate(
       pixel_values.to(device),
       min_length=1,
       bad_words_ids=[[processor.tokenizer.unk_token_id]],
   )

   outputs = model.generate(
       pixel_values.to(device),
       min_length=1,
+      max_new_tokens=500,
       bad_words_ids=[[processor.tokenizer.unk_token_id]],
   )