test

app.py

@@ -11,14 +11,11 @@ tokenizer = MBartTokenizer.from_pretrained(model_name, src_lang="ko_KR", tgt_lan
 
 # prefix = "translate English to German: "
 
-def summarization(News, Summary):
-    summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)
-    summarizer(News, min_length=50, max_length=150)
+summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)
 
-    for result in summarizer(News):
-        print(result)
-        
-        return result["summary_text"]
+def summarization(News, Summary):
+    result = summarizer(News, min_length=50, max_length=150)
+    return result["summary_text"]
 
 if __name__ == '__main__':