wangjin2000
commited on
Commit
•
a305599
1
Parent(s):
62f0a2f
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"
|