Spaces:

ysongsongj
/

my_summarizer

Sleeping

ysongsongj commited on Nov 15, 2023

Commit

26e66bb

1 Parent(s): d4cb847

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ tokenizer = PreTrainedTokenizerFast.from_pretrained(model_name)
 model = BartForConditionalGeneration.from_pretrained(model_name)
 def summ(txt):
-  input_ids = toknizer.encode(txt, return_tensors="pt")
   summary_text_ids = model.generate(
     input_ids = input_ids,
     bos_token_id = model.config.bos_token_id, # BOS는 Beginning Of Sentence

 model = BartForConditionalGeneration.from_pretrained(model_name)
 def summ(txt):
+  input_ids = tokenizer.encode(txt, return_tensors="pt")
   summary_text_ids = model.generate(
     input_ids = input_ids,
     bos_token_id = model.config.bos_token_id, # BOS는 Beginning Of Sentence