File size: 1,084 Bytes
af10332 3a8ce39 43b1092 af10332 3a8ce39 43b1092 3a8ce39 43b1092 3a8ce39 43b1092 af10332 43b1092 af10332 3a8ce39 af10332 43b1092 af10332 43b1092 af10332 43b1092 3a8ce39 af10332 |
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 |
import gradio as gr
import requests
import os
BASE_URL = os.getenv("BASE_URL")
API_KEY = os.getenv("API_KEY")
def generate_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed):
url = f"{BASE_URL}/generate"
headers = {"Authorization": f"Bearer {API_KEY}"}
params = {
"prompt": prompt,
"negative_prompt": negative_prompt,
"sampling_steps": sampling_steps,
"cfg_scale": cfg_scale,
"seed": seed
}
response = requests.get(url, headers=headers, params=params)
if response.status_code == 200:
return response.content
else:
return None
iface = gr.Interface(
fn=generate_image,
inputs=[
gr.Textbox("prompt", label="Prompt"),
gr.Textbox("negative_prompt", label="Negative Prompt"),
gr.Slider(1, 30, 1, label="Sampling Steps", default=10),
gr.Slider(1, 20, 1, label="CFG Scale", default=10),
gr.Textbox("seed", label="Seed (Default: -1)"),
],
outputs=gr.Image(label="Generated Image"),
live=True,
theme="compact"
)
iface.launch()
|