Spaces:
Runtime error
Runtime error
Commit
·
3d765e2
1
Parent(s):
768cb65
remove config
Browse files
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
|
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 |
|