Update app.py

app.py

@@ -1,32 +1,21 @@
-import sys
-import gradio as gr
+from transformers import BlipProcessor, BlipForConditionalGeneration
 from PIL import Image
+import gradio as gr
 
-import vertexai
-from vertexai.preview.vision_models import ImageCaptioningModel #Correct Import
-vertexai.init(project=PROJECT_ID, location=LOCATION)
+processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
+model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
 
 def image_to_prompt(image):
-    """Generates a caption (prompt) from an image using Vertex AI."""
-    if image is None:
-        return "Please upload an image."
-
-    try:
-        model = ImageCaptioningModel.from_pretrained("imagetext@004")  # or imagetext@003
-        captions = model.get_captions(image=image, number_of_results=1)
-        if captions:
-            return captions[0]
-        else:
-            return "Could not generate a caption."
-
-    except Exception as e:
-        return f"An error occurred: {e}"
+    inputs = processor(images=image, return_tensors="pt")
+    outputs = model.generate(**inputs, max_length=50)
+    caption = processor.decode(outputs[0], skip_special_tokens=True)
+    return caption
 
 iface = gr.Interface(
     fn=image_to_prompt,
     inputs=gr.Image(type="pil"),
     outputs=gr.Textbox(label="Generated Prompt"),
-    title="Image to Prompt Generator",
+    title="Image to Prompt Generator (using BLIP)",
     description="Upload an image and get a text prompt describing it.",
 )