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:
|