Update app.py
Browse files
app.py
CHANGED
|
@@ -36,12 +36,12 @@ def img2cap_com(image):
|
|
| 36 |
return translated_caption1,translated_caption2
|
| 37 |
|
| 38 |
inputs = [
|
| 39 |
-
gr.
|
| 40 |
]
|
| 41 |
|
| 42 |
outputs = [
|
| 43 |
-
gr.
|
| 44 |
-
gr.
|
| 45 |
]
|
| 46 |
|
| 47 |
title = "Image Captioning using Pre-trained and Fine-tuned Model"
|
|
|
|
| 36 |
return translated_caption1,translated_caption2
|
| 37 |
|
| 38 |
inputs = [
|
| 39 |
+
gr.Image(type="pil", label="Original Image")
|
| 40 |
]
|
| 41 |
|
| 42 |
outputs = [
|
| 43 |
+
gr.Textbox(label="Caption from pre-trained model"),
|
| 44 |
+
gr.Textbox(label="Caption from fine-tuned model"),
|
| 45 |
]
|
| 46 |
|
| 47 |
title = "Image Captioning using Pre-trained and Fine-tuned Model"
|