| import gradio as gr | |
| from src.inference import CodeStylePipeline | |
| pipe=CodeStylePipeline() | |
| def run(code,style): | |
| return pipe(code,style) | |
| gr.Interface( | |
| fn=run, | |
| inputs=[gr.Textbox(lines=10), gr.Dropdown(['clean','competitive','enterprise','legacy'])], | |
| outputs='text', | |
| title='CodeStyleDiffuser' | |
| ).launch() | |