McCoasta commited on
Commit
269e1df
1 Parent(s): 656d5d6

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +19 -0
  2. requirements.txt +2 -0
app.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from PIL import Image
2
+ import requests
3
+ import gradio as gr
4
+
5
+ from transformers import BlipProcessor, BlipForConditionalGeneration
6
+
7
+ model_id = Salesforceblip-image-captioning-base
8
+
9
+ model = BlipForConditionalGeneration.from_pretrained(model_id)
10
+ processor = BlipProcessor.from_pretrained(model_id)
11
+
12
+ def launch(input)
13
+ image = Image.open(requests.get(input, stream=True).raw).convert('RGB')
14
+ inputs = processor(image, return_tensors=pt)
15
+ out = model.generate(inputs)
16
+ return processor.decode(out[0], skip_special_tokens=True)
17
+
18
+ iface = gr.Interface(launch, inputs=text, outputs=text)
19
+ iface.launch()
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ git+https://github.com/huggingface/transformers.git@main
2
+ torch