evoss's picture
Update app.py
a2a7fd4
raw
history blame contribute delete
394 Bytes
import gradio as gr
from transformers import pipeline
cola = pipeline('text-classification', model='Abirate/bert_fine_tuned_cola')
def correct(x):
correction = cola(x)
return correction
iface = gr.Interface(fn=correct, inputs="text", outputs="text", examples = [
["Is this sentence correkt."],
["I will esteem the country with the biggest economy."]])
iface.launch(debug=True)