Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def predict_step(image_paths):
|
|
31 |
|
32 |
preds = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
|
33 |
preds = [pred.strip() for pred in preds]
|
34 |
-
return
|
35 |
|
36 |
def ImageCaptioning(image):
|
37 |
with tempfile.NamedTemporaryFile(suffix=".png") as temp_image_file:
|
|
|
31 |
|
32 |
preds = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
|
33 |
preds = [pred.strip() for pred in preds]
|
34 |
+
return preds[0]
|
35 |
|
36 |
def ImageCaptioning(image):
|
37 |
with tempfile.NamedTemporaryFile(suffix=".png") as temp_image_file:
|