Pravalika56 commited on
Commit
d228026
·
verified ·
1 Parent(s): 5ee8bbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -9
app.py CHANGED
@@ -1,15 +1,11 @@
1
- import transformers
2
- from transformers import BlipProcessor, BlipFConditionalGeneration
3
  from PIL import Image
 
 
4
 
5
-
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
-
 
1
+ from transformers import BlipProcessor, BlipForConditionalGeneration
 
2
  from PIL import Image
3
+ processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
4
+ model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
5
 
6
+ image = Image.open("/content/blip-imagecaption/skydive.jpg")
7
+ inputs = processor(image, return_tensors = "pt")
 
 
 
 
8
  outputs = model.generate(**inputs)
9
 
10
  caption = processor.decode(outputs[0], skip_special_tokens = True)
11
  print(caption)