Abhishek-D7
commited on
Commit
•
b2c81ec
1
Parent(s):
00a9d47
Update app.py
Browse files
app.py
CHANGED
@@ -29,8 +29,8 @@ def load_translation_models():
|
|
29 |
translation_models, translation_tokenizers = load_translation_models()
|
30 |
|
31 |
def summarize_text(article):
|
32 |
-
inputs = summarization_tokenizer.encode("summarize: " + article, return_tensors="pt", max_length=
|
33 |
-
summary_ids = summarization_model.generate(inputs, max_length=
|
34 |
return summarization_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|
35 |
|
36 |
def translate_text(text, source_lang, target_lang):
|
|
|
29 |
translation_models, translation_tokenizers = load_translation_models()
|
30 |
|
31 |
def summarize_text(article):
|
32 |
+
inputs = summarization_tokenizer.encode("summarize: " + article, return_tensors="pt", max_length=1024, truncation=True)
|
33 |
+
summary_ids = summarization_model.generate(inputs, max_length=150, min_length=30, length_penalty=2.0, num_beams=4, early_stopping=True)
|
34 |
return summarization_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|
35 |
|
36 |
def translate_text(text, source_lang, target_lang):
|