rathapech commited on
Commit
c88106c
·
verified ·
1 Parent(s): 93b5282

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -6,7 +6,15 @@ itt_pipe = pipeline("image-to-text",
6
  model="Salesforce/blip-image-captioning-base")
7
 
8
 
9
- text = itt_pipe(input)
 
 
 
 
 
 
 
 
10
 
11
 
12
  tts_pipe = pipeline("text-to-speech",
 
6
  model="Salesforce/blip-image-captioning-base")
7
 
8
 
9
+ text = gr.Interface(
10
+ fn=itt_pipe,
11
+ inputs=gr.Image(label="Input image",
12
+ type="pil"),
13
+ outputs=gr.Image(label="Text describe the image",
14
+ type="pil")
15
+ )
16
+
17
+ #text = itt_pipe(input)
18
 
19
 
20
  tts_pipe = pipeline("text-to-speech",