Pravalika56 commited on
Commit
036c445
·
verified ·
1 Parent(s): 88df7d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from PIL import Image
5
  def generate_caption(input_image):
6
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
7
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
8
- inputs = processor(image, return_tensors="pt")
9
  outputs = model.generate(**inputs)
10
  caption = processor.decode(outputs[0], skip_special_tokens=True)
11
  return caption
 
5
  def generate_caption(input_image):
6
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
7
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
8
+ inputs = processor(input_image, return_tensors="pt")
9
  outputs = model.generate(**inputs)
10
  caption = processor.decode(outputs[0], skip_special_tokens=True)
11
  return caption