Spaces:
Running
Running
fix: app.py syntax error
Browse files
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 |
)
|