Spaces:
Build error
Build error
| 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() |