Spaces:
Runtime error
Runtime error
File size: 365 Bytes
8231943 1da4a5b 8231943 885bc42 adcdf2c e061e40 8231943 885bc42 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import gradio as gr
from simplet5 import SimpleT5
model = SimpleT5()
# let's load the trained model for inferencing:
model.load_model("byt5","hantech/byt5_correct5", use_gpu=False)
def predict(text):
result = model.predict('correct: '+text)
print(str(result))
return result[0]
gr.Interface(predict, [gr.Textbox(label="Text")], gr.Textbox()).launch()
|