Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,7 +2,7 @@ import torch
|
|
| 2 |
|
| 3 |
import gradio as gr
|
| 4 |
|
| 5 |
-
from transformers import
|
| 6 |
|
| 7 |
import os
|
| 8 |
|
|
@@ -45,7 +45,7 @@ def do_correction(text, model, tokenizer):
|
|
| 45 |
# corrected_sentence = do_correction(sentence, model, tokenizer)
|
| 46 |
|
| 47 |
|
| 48 |
-
gr.Interface(fn=
|
| 49 |
|
| 50 |
|
| 51 |
'''
|
|
|
|
| 2 |
|
| 3 |
import gradio as gr
|
| 4 |
|
| 5 |
+
from transformers import T5ForConditionalGeneration, T5Tokenizer
|
| 6 |
|
| 7 |
import os
|
| 8 |
|
|
|
|
| 45 |
# corrected_sentence = do_correction(sentence, model, tokenizer)
|
| 46 |
|
| 47 |
|
| 48 |
+
gr.Interface(fn=do_correction, inputs="text", outputs="text")
|
| 49 |
|
| 50 |
|
| 51 |
'''
|