gabrielwithhappy commited on
Commit
836aaa3
2 Parent(s): 51a4295 d050d19

fix missing input

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -17,12 +17,17 @@ def get_summary(input_text):
17
  summary_model = BartForConditionalGeneration.from_pretrained("ainize/kobart-news")
18
  input_ids = tokenizer.encode(input_text, return_tensors="pt")
19
  summary_text_ids = summary_model.generate(
 
20
  input_ids=input_ids,
21
  length_penalty=2,
22
  top_p=0.9,
 
 
 
 
23
  max_length=128,
24
  min_length=12,
25
- num_beams=4,
26
  )
27
  # "task_specific_params": {
28
  # "summarization": {
 
17
  summary_model = BartForConditionalGeneration.from_pretrained("ainize/kobart-news")
18
  input_ids = tokenizer.encode(input_text, return_tensors="pt")
19
  summary_text_ids = summary_model.generate(
20
+ <<<<<<< HEAD
21
  input_ids=input_ids,
22
  length_penalty=2,
23
  top_p=0.9,
24
+ =======
25
+ length_penalty=1,
26
+ # top_p=0.9,
27
+ >>>>>>> d050d191bf6ac2aad382b3d7ee3b1fe6e6302502
28
  max_length=128,
29
  min_length=12,
30
+ num_beams=2,
31
  )
32
  # "task_specific_params": {
33
  # "summarization": {