coder160/cpu_machine_learner

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