wangjin2000 commited on
Commit
a305599
1 Parent(s): 62f0a2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -36,12 +36,12 @@ def img2cap_com(image):
36
  return translated_caption1,translated_caption2
37
 
38
  inputs = [
39
- gr.inputs.Image(type="pil", label="Original Image")
40
  ]
41
 
42
  outputs = [
43
- gr.outputs.Textbox(label="Caption from pre-trained model"),
44
- gr.outputs.Textbox(label="Caption from fine-tuned model"),
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"