Update app.py
Browse files
app.py
CHANGED
|
@@ -55,7 +55,7 @@ def process_image(image):
|
|
| 55 |
)[0]
|
| 56 |
caption = processor.post_process_generation(
|
| 57 |
generated_text, task="<CAPTION>", image_size=(image.width, image.height)
|
| 58 |
-
)
|
| 59 |
return caption
|
| 60 |
|
| 61 |
except Exception as e:
|
|
|
|
| 55 |
)[0]
|
| 56 |
caption = processor.post_process_generation(
|
| 57 |
generated_text, task="<CAPTION>", image_size=(image.width, image.height)
|
| 58 |
+
)[0]
|
| 59 |
return caption
|
| 60 |
|
| 61 |
except Exception as e:
|