Spaces:
Sleeping
Sleeping
File size: 882 Bytes
dca9cd6 8e967a1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import gradio as gr
from transformers import pipeline
def correct_text(text, temperature=1.0, top_p=0.9, top_k=50):
model_id = "grammarly/coedit-large"
pipe = pipeline("text2text-generation", model=model_id,
generation_kwargs={"temperature": temperature, "top_p": top_p, "top_k": top_k})
corrected = pipe(text)[0]['generated_text']
return corrected
interface = gr.Interface(fn=correct_text,
inputs=[gr.inputs.TextArea(label="Input Text"),
gr.inputs.Slider(minimum=0.1, maximum=1.0, default=1.0, label="Temperature"),
gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.9, label="Top_p"),
gr.inputs.Slider(minimum=0, maximum=100, default=50, label="Top_k")],
outputs="text")
interface.launch()
|