File size: 323 Bytes
b2edbd0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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()
|