Fateh Singh commited on
Commit
3cde9b2
·
1 Parent(s): ebe7359

Add application file

Browse files
Files changed (1) hide show
  1. app.py +19 -0
app.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import BlipProcessor, BlipForConditionalGeneration
3
+ from PIL import Image
4
+ import requests
5
+ import io
6
+
7
+ processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
8
+ model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
9
+
10
+ def caption_image(image):
11
+ inputs = processor(images=image, return_tensors="pt")
12
+ out = model.generate(**inputs)
13
+ caption = processor.decode(out[0], skip_special_tokens=True)
14
+ return caption
15
+
16
+ gr.Interface(fn=caption_image,
17
+ inputs=gr.Image(type="pil"),
18
+ outputs="text",
19
+ title="Image Caption Generator").launch()