Update app.py

app.py

@@ -1,18 +1,12 @@
 import gradio as gr
-
 from transformers import PreTrainedTokenizerFast, BartForConditionalGeneration
-# from transformers import로 시작하는 import 문을 보면
-# 많은 경우 AutoTokenizer, AutoModel
-# tokenizer = AutoTokenizer.from_pretrained("model 이름 어쩌고 저쩌고")
-# PreTrainedTokenizerFast : https://huggingface.co/docs/transformers/main_classes/tokenizer
-# BART는 encoder-decoder 모델의 예시
 
 model_name = "ainize/kobart-news"
 tokenizer = PreTrainedTokenizerFast.from_pretrained(model_name)
 model = BartForConditionalGeneration.from_pretrained(model_name)
 
 # 원문을 받아서 요약문을 반환
-def summ(txt):
+def summ(input_text):  # 매개변수명을 txt에서 input_text로 변경
   input_ids = tokenizer.encode(input_text, return_tensors="pt")
   summary_text_ids = model.generate(
     input_ids=input_ids,
@@ -28,4 +22,4 @@ interface = gr.Interface(summ,
                         [gr.Textbox(label="original text")],
                         [gr.Textbox(label="summary")])
 
-interface.launch()
+interface.launch()