Update app.py

app.py

@@ -10,8 +10,13 @@ model_id = "Salesforce/blip-image-captioning-base"
 model = BlipForConditionalGeneration.from_pretrained(model_id)
 processor = BlipProcessor.from_pretrained(model_id)
 
-def launch(input_image):
-    image = Image.open(BytesIO(input_image)).convert('RGB')
+def decode_base64_image(image_string):
+  base64_image = base64.b64decode(image_string)
+  buffer = BytesIO(base64_image)
+  return  Image.open(buffer)
+
+def launch(input_b64):
+    image = decode_base64_image(input_b64)
     #inputsData = data.pop("inputs", data)
     # decode base64 image to PIL
     inputs = processor(image, return_tensors="pt")