Spaces:
Runtime error
Runtime error
File size: 1,473 Bytes
d12f2bb 9fca7c3 d12f2bb 279af2f d12f2bb b8c2f1b d12f2bb a73b3bc d12f2bb 19d53bf d12f2bb 9fca7c3 d12f2bb 9fca7c3 d12f2bb 9fca7c3 d12f2bb 427cb66 d12f2bb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
import os
import gradio as gr
from google import genai
from google.genai import types
# Set your Google API key
os.environ["GOOGLE_API_KEY"] = "AIzaSyDL5Rilo7ptJpUOZdY6wy8PJYUcVcnDADs"
# Initialize the Google Generative AI client
client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
MODEL_NAME = "gemini-2.5-flash-image-preview"
def remix_images(image1, image2, prompt):
try:
# Run request to Google Generative AI
response = client.models.generate_images(
model=MODEL_NAME,
prompt=prompt,
images=[image1, image2]
)
# Extract the first generated image
if response.generated_images and len(response.generated_images) > 0:
img = response.generated_images[0]
return img.image # returns PIL.Image
else:
return None
except Exception as e:
return f"Error: {str(e)}"
# Gradio UI
with gr.Blocks() as demo:
gr.Markdown("## 🖼️ Google Gemini 2.5 Flash Image Preview (via Gradio)")
with gr.Row():
img1 = gr.Image(type="filepath", label="Upload Image 1")
img2 = gr.Image(type="filepath", label="Upload Image 2")
prompt = gr.Textbox(label="Prompt", placeholder="Describe how to remix the images...")
btn = gr.Button("Generate")
output = gr.Image(label="Generated Image")
btn.click(fn=remix_images, inputs=[img1, img2, prompt], outputs=output)
if __name__ == "__main__":
demo.launch()
|