File size: 394 Bytes
201e647
 
a2a7fd4
 
320960c
201e647
 
 
320960c
201e647
 
 
1
2
3
4
5
6
7
8
9
10
11
12
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)