JayGilDS commited on
Commit
a48549e
·
verified ·
1 Parent(s): d0b6b6f

Update app.py

Browse files

Changed GUI labels

Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -1,16 +1,16 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- pipe = pipeline("image-to-text",
5
- model="Salesforce/blip-image-captioning-base")
6
-
7
 
8
  def launch(input):
9
  out = pipe(input)
10
  return out[0]['generated_text']
11
 
12
- iface = gr.Interface(launch,
13
- inputs=gr.Image(type='pil'),
14
- outputs="text")
 
 
15
 
16
  iface.launch()
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ pipe = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
 
 
5
 
6
  def launch(input):
7
  out = pipe(input)
8
  return out[0]['generated_text']
9
 
10
+ iface = gr.Interface(
11
+ fn=launch,
12
+ inputs=gr.Image(type='pil', label="Upload a picture to be captioned"),
13
+ outputs=gr.Textbox(label="Picture Caption:")
14
+ )
15
 
16
  iface.launch()