Hyeonsieun commited on
Commit
936ce22
·
verified ·
1 Parent(s): 04a91ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import torch
2
 
3
  import gradio as gr
4
 
5
- from transformers import pipeline, T5ForConditionalGeneration, T5Tokenizer
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=yt_do_correction, inputs="text", outputs="text")
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
  '''