File size: 1,207 Bytes
af10332 3a8ce39 43b1092 af10332 3a8ce39 43b1092 70ba535 3a8ce39 70ba535 af10332 43b1092 70ba535 43b1092 af10332 3a8ce39 af10332 70ba535 af10332 70ba535 af10332 3a8ce39 70ba535 |
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 |
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):
headers = {
'Authorization': f'Bearer {API_KEY}',
}
data = {
'prompt': prompt,
'negative_prompt': negative_prompt,
'sampling_steps': sampling_steps,
'cfg_scale': cfg_scale,
'seed': seed,
}
response = requests.post(BASE_URL, headers=headers, json=data)
if response.status_code == 200:
image_data = response.json()['image']
return image_data
else:
return None
iface = gr.Interface(
fn=generate_image,
inputs=[
gr.Textbox('text', label='Prompt', default='Enter your prompt...'),
gr.Textbox('text', label='Negative Prompt', default='Enter your negative prompt...'),
],
outputs=gr.Image(label='Generated Image'),
live=True,
)
with iface.launch(debug=True):
iface.add_slider('Sampling Steps', 1, 30, 1)
iface.add_slider('CFG Scale', 1, 20, 1)
iface.add_textbox('Seed', default='-1')
iface.add_button('Generate', None)
iface.add_download_button('Download Image')
|