Pravalika56 commited on
Commit
b488b6f
·
verified ·
1 Parent(s): 4bbcc7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -6,4 +6,10 @@ from PIL import Image
6
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
7
  model = BlipFConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
8
 
9
- our_image = Image.open("")
 
 
 
 
 
 
 
6
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
7
  model = BlipFConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
8
 
9
+ our_image = Image.open("elephant.jpg")
10
+ inputs = processor(our_image, return_tensors = "pt")
11
+ outputs = model.generate(**inputs)
12
+
13
+ caption = processor.decode(outputs[0], skip_special_tokens = True)
14
+ print(caption)
15
+