memegpt commited on
Commit
32bb55d
·
1 Parent(s): c5e708d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -10,11 +10,10 @@ model_id = "Salesforce/blip-image-captioning-base"
10
  model = BlipForConditionalGeneration.from_pretrained(model_id)
11
  processor = BlipProcessor.from_pretrained(model_id)
12
 
13
- def launch(image):
14
- #image = Image.open(BytesIO(input_image)).convert('RGB')
15
- inputsData = data.pop("inputs", data)
16
  # decode base64 image to PIL
17
- image = Image.open(BytesIO(base64.b64decode(inputsData["image"])))
18
  inputs = processor(image, return_tensors="pt")
19
  out = model.generate(**inputs)
20
  return processor.decode(out[0], skip_special_tokens=True)
 
10
  model = BlipForConditionalGeneration.from_pretrained(model_id)
11
  processor = BlipProcessor.from_pretrained(model_id)
12
 
13
+ def launch(input_image):
14
+ image = Image.open(BytesIO(input_image)).convert('RGB')
15
+ #inputsData = data.pop("inputs", data)
16
  # decode base64 image to PIL
 
17
  inputs = processor(image, return_tensors="pt")
18
  out = model.generate(**inputs)
19
  return processor.decode(out[0], skip_special_tokens=True)