Spaces:
Runtime error
Runtime error
File size: 2,735 Bytes
254af27 89cd2fd 254af27 89cd2fd |
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 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
import gradio as gr
from PIL import Image
import torch
from diffusers import StableDiffusionPipeline
# Load the model
model_id = "models/black-forest-labs/FLUX.1-dev"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")
def generate_image(prompt, negative_prompt, num_inference_steps, guidance_scale, width, height):
image = pipe(
prompt=prompt,
negative_prompt=negative_prompt,
num_inference_steps=num_inference_steps,
guidance_scale=guidance_scale,
width=width,
height=height
).images[0]
return image
def edit_image(image, brightness, contrast, saturation):
edited = Image.fromarray(image)
edited = edited.convert('RGB')
edited = edited.point(lambda p: p * brightness)
edited = edited.point(lambda p: 128 + (p - 128) * contrast)
edited = edited.convert('HSV')
h, s, v = edited.split()
s = s.point(lambda p: p * saturation)
edited = Image.merge('HSV', (h, s, v)).convert('RGB')
return edited
with gr.Blocks(css="style.css") as demo:
gr.Markdown("# FLUX.1 Image Generator")
with gr.Tab("Generate"):
with gr.Row():
with gr.Column():
prompt = gr.Textbox(label="Prompt")
negative_prompt = gr.Textbox(label="Negative Prompt")
steps = gr.Slider(minimum=1, maximum=100, value=50, step=1, label="Inference Steps")
guidance = gr.Slider(minimum=1, maximum=20, value=7.5, step=0.1, label="Guidance Scale")
width = gr.Slider(minimum=256, maximum=1024, value=512, step=64, label="Width")
height = gr.Slider(minimum=256, maximum=1024, value=512, step=64, label="Height")
generate_btn = gr.Button("Generate")
with gr.Column():
output = gr.Image(label="Generated Image")
with gr.Tab("Edit"):
with gr.Row():
with gr.Column():
input_image = gr.Image(label="Input Image")
brightness = gr.Slider(minimum=0.5, maximum=1.5, value=1.0, step=0.1, label="Brightness")
contrast = gr.Slider(minimum=0.5, maximum=1.5, value=1.0, step=0.1, label="Contrast")
saturation = gr.Slider(minimum=0.5, maximum=1.5, value=1.0, step=0.1, label="Saturation")
edit_btn = gr.Button("Apply Edits")
with gr.Column():
edited_output = gr.Image(label="Edited Image")
generate_btn.click(generate_image, inputs=[prompt, negative_prompt, steps, guidance, width, height], outputs=output)
edit_btn.click(edit_image, inputs=[input_image, brightness, contrast, saturation], outputs=edited_output)
demo.launch() |