Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ tokenizer = MarianTokenizer.from_pretrained(model_name)
|
|
| 9 |
model = MarianMTModel.from_pretrained(model_name)
|
| 10 |
|
| 11 |
def traducir(texto):
|
| 12 |
-
text_summarized = summarizer(texto, max_length=
|
| 13 |
texto_solo = ">>esp<< " + text_summarized[0]["summary_text"]
|
| 14 |
translated_entero = model.generate(**tokenizer(texto, return_tensors="pt", padding=True))
|
| 15 |
texto_entero = [tokenizer.decode(t, skip_special_tokens=True) for t in translated_entero]
|
|
|
|
| 9 |
model = MarianMTModel.from_pretrained(model_name)
|
| 10 |
|
| 11 |
def traducir(texto):
|
| 12 |
+
text_summarized = summarizer(texto, max_length=1000000, min_length=30, do_sample=False)
|
| 13 |
texto_solo = ">>esp<< " + text_summarized[0]["summary_text"]
|
| 14 |
translated_entero = model.generate(**tokenizer(texto, return_tensors="pt", padding=True))
|
| 15 |
texto_entero = [tokenizer.decode(t, skip_special_tokens=True) for t in translated_entero]
|