Create app.py

app.py

@@ -0,0 +1,29 @@
+import sys
+import gradio as gr
+from PIL import Image
+
+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)  # Get the top caption
+        if captions:
+            return captions[0]
+        else:
+            return "Could not generate a caption."
+
+    except Exception as e:
+        return f"An error occurred: {e}"
+
+iface = gr.Interface(
+    fn=image_to_prompt,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Textbox(label="Generated Prompt"),
+    title="Image to Prompt Generator",
+    description="Upload an image and get a text prompt describing it.",
+)
+
+iface.launch()