Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
-from transformers import PreTrainedTokenizerFast, BartforConditionalGeneration
+from transformers import PreTrainedTokenizerFast, BartForConditionalGeneration
 
 model_name = "ainize/kobart-news"
 tokenizer = PreTrainedTokenizerFast.from_pretrained(model_name)
-model = BartforConditionalGeneration.from_pretrained(model_name)
+model = BartForConditionalGeneration.from_pretrained(model_name)
 
 def summ(txt):
   input_ids = tokenizer.encode(input_text, return_tensors="pt")