File size: 919 Bytes
594459e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from transformers import pipeline as Pipe
class Traductor:
def EN_ES(texto:str)->str:
_traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-en-es")
_traduccion = _traductor(texto)[0]
return _traduccion.get('generated_text')
def ES_EN(texto:str)->str:
_traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-es-en")
_traduccion = _traductor(texto)[0]
return _traduccion.get('generated_text')
def AR_ES(texto:str)->str:
_traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-ar-es")
_traduccion = _traductor(texto)[0]
return _traduccion.get('generated_text')
class Abstractor:
def resumen(texto:str)->str:
_abstractor = Pipe("text2text-generation", model="facebook/bart-large-cnn")
_resumen = _abstractor(texto)[0]
return _resumen.get('generated_text') |