Minor update

app.py

@@ -8,7 +8,8 @@ import gradio as gr
 
 hf_api_key = os.environ['HF_API_KEY']
 
-get_completion = pipeline("ner", model="Salesforce/blip-image-captioning-base")
+# Load the image-to-text pipeline with BLIP model
+get_completion = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
 
 def image_to_base64_str(pil_image):
     byte_arr = io.BytesIO()
@@ -18,21 +19,20 @@ def image_to_base64_str(pil_image):
     return str(base64.b64encode(byte_arr).decode('utf-8'))
 
 def captioner(image):
-    base64_image = image_to_base64_str(image)
-    result = get_completion(base64_image)
+    # The BLIP model expects a PIL image directly
+    result = get_completion(image)
 
     return result[0]['generated_text']
 
-gr.close_all()
 demo = gr.Interface(fn=captioner,
                     inputs=[gr.Image(label="Upload image", type="pil")],
                     outputs=[gr.Textbox(label="Caption")],
                     title="Image Captioning with BLIP",
                     description="Caption any image using the BLIP model",
-                    allow_flagging="never",
+                    flagging_mode="never",  # Updated from allow_flagging
                     examples=["images/christmas_dog.jpg", "images/bird_flight.jpg", "images/cow.jpg"])
 
 demo.launch(
     share=True, 
-    #server_port=int(os.environ['PORT1'])
+    # server_port=int(os.environ.get('PORT3', 7860))  # Uncomment if needed
 )