gabrielwithhappy commited on
Commit
3d765e2
·
1 Parent(s): 768cb65

remove config

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -18,12 +18,6 @@ def get_summary(input_text):
18
  input_ids = tokenizer.encode(input_text, return_tensors="pt")
19
  summary_text_ids = summary_model.generate(
20
  input_ids=input_ids,
21
- bos_token_id=summary_model.config.bos_token_id,
22
- eos_token_id=summary_model.config.eos_token_id,
23
- length_penalty=2.0,
24
- max_length=142,
25
- min_length=56,
26
- num_beams=4,
27
  )
28
  return tokenizer.decode(summary_text_ids[0], skip_special_tokens=True)
29
 
@@ -62,7 +56,7 @@ class news_collector:
62
 
63
  for news_url in news_links:
64
  article = self.get_new_parser(news_url)
65
- self.examples_text.append([get_summary(article.text[:1000]), news_url])
66
 
67
 
68
 
 
18
  input_ids = tokenizer.encode(input_text, return_tensors="pt")
19
  summary_text_ids = summary_model.generate(
20
  input_ids=input_ids,
 
 
 
 
 
 
21
  )
22
  return tokenizer.decode(summary_text_ids[0], skip_special_tokens=True)
23
 
 
56
 
57
  for news_url in news_links:
58
  article = self.get_new_parser(news_url)
59
+ self.examples_text.append([get_summary(article.text), news_url])
60
 
61
 
62