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()