Abhishek-D7 commited on
Commit
b2c81ec
1 Parent(s): 00a9d47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=6000, truncation=True)
33
- summary_ids = summarization_model.generate(inputs, max_length=256, min_length=10, 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):
 
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):