ArtemZvagin's picture
Upload 2 files
3f15bce verified
raw
history blame contribute delete
776 Bytes
import json
import gradio as gr
with open('all_styles.json', encoding='utf-8') as file:
styles = json.load(file)
all_styles = list(styles.keys())
def apply_style(prompt, style):
style_prompt, neg_prompt = styles.get(style, (None, None))
if style_prompt:
prompt = style_prompt.replace('{prompt}', prompt)
return prompt, neg_prompt
else:
return prompt, ''
demo = gr.Interface(
fn=apply_style,
inputs=[gr.Text(value='', placeholder='your prompt'), gr.Dropdown(label="Style", choices=all_styles, value=all_styles[0])],
outputs=[gr.Textbox(label='Prompt', show_copy_button=True), gr.Textbox(label='Negative Prompt', show_copy_button=True)],
title='Stylize your prompt',
allow_flagging='never'
)
demo.launch()