Update app.py
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ def generate_image_captions(image, text):
|
|
| 29 |
max_new_tokens=50
|
| 30 |
)
|
| 31 |
|
| 32 |
-
prediction_text = processor.decode(prediction[0],
|
| 33 |
return prediction_text
|
| 34 |
|
| 35 |
|
|
@@ -51,7 +51,7 @@ demo = gr.Interface(
|
|
| 51 |
outputs="text",
|
| 52 |
examples=[
|
| 53 |
[os.path.join(os.getcwd(), 'two_bear.png'), ""],
|
| 54 |
-
[os.path.join(os.getcwd(), 'three_women.png'), "What is the woman in the middle's dress's color
|
| 55 |
[os.path.join(os.getcwd(), 'cat_with_food.png'), "Describe the picture:"],
|
| 56 |
[os.path.join(os.getcwd(), 'dog_with_frisbee.png'), "What is the color of the frisbee in the photo? Answer:"],
|
| 57 |
[os.path.join(os.getcwd(), 'stop_sign.png'), "What does the sign in the picture say? Answer:"]
|
|
|
|
| 29 |
max_new_tokens=50
|
| 30 |
)
|
| 31 |
|
| 32 |
+
prediction_text = processor.decode(prediction[0], skip_special_tokens=True)
|
| 33 |
return prediction_text
|
| 34 |
|
| 35 |
|
|
|
|
| 51 |
outputs="text",
|
| 52 |
examples=[
|
| 53 |
[os.path.join(os.getcwd(), 'two_bear.png'), ""],
|
| 54 |
+
[os.path.join(os.getcwd(), 'three_women.png'), "What is the woman in the middle's dress's color? Answer:"],
|
| 55 |
[os.path.join(os.getcwd(), 'cat_with_food.png'), "Describe the picture:"],
|
| 56 |
[os.path.join(os.getcwd(), 'dog_with_frisbee.png'), "What is the color of the frisbee in the photo? Answer:"],
|
| 57 |
[os.path.join(os.getcwd(), 'stop_sign.png'), "What does the sign in the picture say? Answer:"]
|