Spaces:

hantech
/

byt5_correct

Runtime error

hantech commited on Sep 6, 2023

Commit

adcdf2c

1 Parent(s): 885bc42

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ model = SimpleT5()
 model.load_model("byt5","hantech/byt5_correct2", use_gpu=False)
 def predict(text):
-    result = model.predict(text)
     return result[0]
 gr.Interface(predict, [gr.Textbox(label="Text")], gr.Textbox()).launch()

 model.load_model("byt5","hantech/byt5_correct2", use_gpu=False)
 def predict(text):
+    result = model.predict('correct: '+text)
     return result[0]
 gr.Interface(predict, [gr.Textbox(label="Text")], gr.Textbox()).launch()