Update app.py

app.py

@@ -10,8 +10,10 @@ model, vis_processors, _ = load_model_and_preprocess(name="blip_caption", model_
 
 # Define the input and output functions for Gradio
 def generate_caption(image_file):
-        # Load the image from the file path
-        raw_image = Image.open(image_file).convert("RGB")
+        image = cv2.imdecode(image_file.read(), cv2.IMREAD_COLOR)
+
+        # Preprocess the image using the Blip-Caption model's visual processors
+        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
         # Preprocess the image using the Blip-Caption model's visual processors
         image = vis_processors["eval"](raw_image).unsqueeze(0)