astrojuanlu commited on
Commit
3d2b507
1 Parent(s): 08a5bb7

Fix model name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ def summarize(post):
8
  )
9
  lang_results = detector(post)
10
  if (detected_lang := lang_results[0]["label"]) == "en":
11
- summarizer_model_name = "papluca/xlm-roberta-base-language-detection"
12
  elif detected_lang == "es":
13
  summarizer_model_name = "ELiRF/mbart-large-cc25-dacsa-es"
14
  pipe = pipeline("summarization", model=summarizer_model_name)
 
8
  )
9
  lang_results = detector(post)
10
  if (detected_lang := lang_results[0]["label"]) == "en":
11
+ summarizer_model_name = "sshleifer/distilbart-cnn-12-6"
12
  elif detected_lang == "es":
13
  summarizer_model_name = "ELiRF/mbart-large-cc25-dacsa-es"
14
  pipe = pipeline("summarization", model=summarizer_model_name)