tonyassi commited on
Commit
a29c427
β€’
1 Parent(s): 3a2edfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,12 +6,12 @@ from transformers import BlipProcessor, BlipForConditionalGeneration
6
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
7
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
8
 
9
- def caption(img, min, max):
10
  raw_image = Image.open(img).convert('RGB')
11
 
12
  inputs = processor(raw_image, return_tensors="pt")
13
 
14
- out = model.generate(**inputs, min_length=min, max_length=max)
15
  return processor.decode(out[0], skip_special_tokens=True)
16
 
17
  def greet(img):
 
6
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
7
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
8
 
9
+ def caption(img, min_, max_):
10
  raw_image = Image.open(img).convert('RGB')
11
 
12
  inputs = processor(raw_image, return_tensors="pt")
13
 
14
+ out = model.generate(**inputs, min_length=min_, max_length=max_)
15
  return processor.decode(out[0], skip_special_tokens=True)
16
 
17
  def greet(img):