File size: 1,471 Bytes
31166da |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
from transformers import pipeline as Pipe
class Traductor:
def EN_ES(texto:str)->str:
try:
_traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-en-es")
_traduccion = _traductor(texto)[0]
_response = _traduccion.get('generated_text')
except Exception as e:
_response = str(e)
finally:
return _response
def ES_EN(texto:str)->str:
try:
_traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-es-en")
_traduccion = _traductor(texto)[0]
_response = _traduccion.get('generated_text')
except Exception as e:
_response = str(e)
finally:
return _response
def AR_ES(texto:str)->str:
try:
_traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-ar-es")
_traduccion = _traductor(texto)[0]
_response = _traduccion.get('generated_text')
except Exception as e:
_response = str(e)
finally:
return _response
class Abstractor:
def resumen(texto:str)->str:
try:
_abstractor = Pipe("text2text-generation", model="facebook/bart-large-cnn")
_resumen = _abstractor(texto)[0]
_response = _resumen.get('generated_text')
except Exception as e:
_response = str(e)
finally:
return _response |