davanstrien HF Staff commited on
Commit
275bb85
·
1 Parent(s): e1b1045

Increase max_new_tokens in HF_PIPE for enhanced prediction capacity

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ def predict(pil_image):
89
  ]
90
 
91
  # Use the pipeline with the properly formatted messages
92
- return HF_PIPE(messages)
93
 
94
  def run_hf_ocr(image_path):
95
  """
 
89
  ]
90
 
91
  # Use the pipeline with the properly formatted messages
92
+ return HF_PIPE(messages,max_new_tokens=8096)
93
 
94
  def run_hf_ocr(image_path):
95
  """