airzy1 commited on
Commit
5be1a48
·
verified ·
1 Parent(s): ee5eab8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -135,7 +135,7 @@ def analyze_pantry(image: Image.Image) -> Tuple[Image.Image, Dict[str, Any]]:
135
  with torch.inference_mode():
136
  output_ids = model.generate(
137
  **inputs,
138
- max_new_tokens=512,
139
  do_sample=False,
140
  )
141
 
 
135
  with torch.inference_mode():
136
  output_ids = model.generate(
137
  **inputs,
138
+ max_new_tokens=1024,
139
  do_sample=False,
140
  )
141