test

app.py

@@ -13,9 +13,9 @@ tokenizer = MBartTokenizer.from_pretrained(model_name, src_lang="ko_KR", tgt_lan
 
 def summarization(News, Summary):
     summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)
-    summarizer("An apple a day, keeps the doctor away", min_length=50, max_length=150)
+    summarizer(News, min_length=50, max_length=150)
 
-    for result in summarizer(Summary):
+    for result in summarizer(News):
         print(result)
     return result
 
@@ -24,8 +24,8 @@ if __name__ == '__main__':
     #Create a gradio app with a button that calls predict()
     app = gr.Interface(
         fn=summarization,
-        inputs='text',
-        outputs='text', 
+        inputs=gr.inputs.Textbox(lines=10, label="News"),
+        outputs=gr.outputs.Textbox(lines=10, label="Summary"), 
         title="한국어 뉴스 요약 생성기",
         description="Korean News Summary Generator"
         )