Update app.py
Browse files
app.py
CHANGED
|
@@ -2,27 +2,24 @@ import gradio as gr
|
|
| 2 |
from huggingface_hub import InferenceClient
|
| 3 |
|
| 4 |
# Define the inference function
|
| 5 |
-
def generate_image(prompt, seed
|
| 6 |
client = InferenceClient(model="tryonlabs/FLUX.1-dev-LoRA-Lehenga-Generator")
|
| 7 |
-
# Assuming the model accepts these parameters; adjust based on actual API
|
| 8 |
image = client.text_to_image(
|
| 9 |
prompt=prompt,
|
| 10 |
-
seed=seed if seed >= 0 else None
|
| 11 |
-
denoise=denoise # Denoising strength, typically between 0 and 1
|
| 12 |
)
|
| 13 |
return image
|
| 14 |
|
| 15 |
-
# Create the Gradio interface
|
| 16 |
interface = gr.Interface(
|
| 17 |
fn=generate_image,
|
| 18 |
inputs=[
|
| 19 |
gr.Textbox(label="Prompt", placeholder="Enter your prompt here..."),
|
| 20 |
-
gr.Slider(label="Seed", minimum=-1, maximum=1000000, step=1, value=-1, info="Set to -1 for random seed")
|
| 21 |
-
gr.Slider(label="Denoise Strength", minimum=0.0, maximum=1.0, step=0.01, value=0.8, info="Controls noise reduction (0 to 1)")
|
| 22 |
],
|
| 23 |
outputs=gr.Image(label="Generated Lehenga Image"),
|
| 24 |
title="FLUX.1-dev Lehenga Generator",
|
| 25 |
-
description="Generate custom Lehenga designs using FLUX.1-dev with LoRA. Adjust seed
|
| 26 |
)
|
| 27 |
|
| 28 |
# Launch the interface
|
|
|
|
| 2 |
from huggingface_hub import InferenceClient
|
| 3 |
|
| 4 |
# Define the inference function
|
| 5 |
+
def generate_image(prompt, seed):
|
| 6 |
client = InferenceClient(model="tryonlabs/FLUX.1-dev-LoRA-Lehenga-Generator")
|
|
|
|
| 7 |
image = client.text_to_image(
|
| 8 |
prompt=prompt,
|
| 9 |
+
seed=seed if seed >= 0 else None # Use -1 for random seed
|
|
|
|
| 10 |
)
|
| 11 |
return image
|
| 12 |
|
| 13 |
+
# Create the Gradio interface
|
| 14 |
interface = gr.Interface(
|
| 15 |
fn=generate_image,
|
| 16 |
inputs=[
|
| 17 |
gr.Textbox(label="Prompt", placeholder="Enter your prompt here..."),
|
| 18 |
+
gr.Slider(label="Seed", minimum=-1, maximum=1000000, step=1, value=-1, info="Set to -1 for random seed")
|
|
|
|
| 19 |
],
|
| 20 |
outputs=gr.Image(label="Generated Lehenga Image"),
|
| 21 |
title="FLUX.1-dev Lehenga Generator",
|
| 22 |
+
description="Generate custom Lehenga designs using FLUX.1-dev with LoRA. Adjust seed for control.",
|
| 23 |
)
|
| 24 |
|
| 25 |
# Launch the interface
|