Spaces:
Runtime error
Runtime error
astrojuanlu
commited on
Commit
·
3d2b507
1
Parent(s):
08a5bb7
Fix model name
Browse files
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 = "
|
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)
|