Upload 3 files

README.md

@@ -1,13 +1,12 @@
 ---
-title: AI Image Captioner Multilingual
-emoji: 📚
+title: AI Image Captioner MultiLang
+emoji: 🐠
 colorFrom: pink
-colorTo: purple
+colorTo: green
 sdk: gradio
-sdk_version: 5.25.0
+sdk_version: 5.24.0
 app_file: app.py
 pinned: false
-license: apache-2.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,47 @@
+import google.generativeai as genai
+import gradio as gr
+from PIL import Image
+import os
+
+# Set up Gemini API key
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))  # Replace with your API key
+
+def generate_poem(image, prompt, language):
+    """Generate a poem based on the image and prompt, optimized for speed."""
+    if not language:
+        return "Please select a language."
+
+    model = genai.GenerativeModel("gemini-1.5-flash")  # Use a faster model
+
+    # Convert and resize the image for faster processing
+    img = image.convert("RGB")
+    img = img.resize((256, 256))  
+
+    # Optimized short prompt
+    full_prompt = f"Generate a short poem in {language} based on this image and theme: {prompt}."
+
+    # Generate poem without streaming
+    response = model.generate_content([img, full_prompt])  
+    output_text = response.text
+
+    return output_text
+
+# Gradio UI
+iface = gr.Interface(
+    fn=generate_poem,
+    inputs=[
+        gr.Image(type="pil"),  # Directly loads as PIL object
+        gr.Textbox(label="Enter a theme for the poem (in English)"), 
+        gr.Dropdown(
+            ["Hindi", "Tamil", "Telugu", "Malayalam", "Kannada", "Marathi", "Bengali"], 
+            label="Select Output Language"
+        )
+    ],
+    outputs="text",
+    title="Multilingual Image Poetry Generator",
+    description="Upload an image, enter a theme in English, and get a poem in your chosen regional language."
+)
+
+# Run the app
+if __name__ == "__main__":
+    iface.launch()

gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text