kwyong commited on
Commit
8ae7ec8
·
1 Parent(s): f1b497b

fix: app.py syntax error

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,8 +21,8 @@ def predict(img):
21
  pixel_values = processor(img, return_tensors="pt").pixel_values
22
 
23
  outputs = model.generate(
24
- pixel_values.to(device)
25
- min_length=1
26
  max_new_tokens=30,
27
  bad_words_ids=[[processor.tokenizer.unk_token_id]],
28
  )
 
21
  pixel_values = processor(img, return_tensors="pt").pixel_values
22
 
23
  outputs = model.generate(
24
+ pixel_values.to(device),
25
+ min_length=1,
26
  max_new_tokens=30,
27
  bad_words_ids=[[processor.tokenizer.unk_token_id]],
28
  )