Update app.py

app.py

@@ -1,3 +1,18 @@
+import torch
+
+import gradio as gr
+
+from transformers import T5ForConditionalGeneration, T5Tokenizer
+
+import os
+
+#import whisper
+
+import matplotlib as plt
+
+# whisper_model = whisper.load_model('large-v2')      # Whisper 모델을 불러오기
+
+
 path = "Hyeonsieun/NTtoGT_1epoch"
 tokenizer = T5Tokenizer.from_pretrained(path)
 model = T5ForConditionalGeneration.from_pretrained(path)
@@ -28,4 +43,8 @@ def do_correction(text):
     return corrected_sentence
 
 
+
+
+
+
 gr.Interface(fn=do_correction, inputs="text", outputs="text").launch()