from transformers import pipeline | |
from transformers.pipelines.base import Pipeline | |
class Interpretador: | |
def __init__(self, texto: str, pergunta: str) -> None: | |
self._texto = texto | |
self._pergunta = pergunta | |
self._responder() | |
def resposta(self) -> str: | |
return self._resposta | |
def _responder(self): | |
resposta_pipeline = pipeline("question-answering", model="pierreguillou/bert-large-cased-squad-v1.1-portuguese") | |
resposta_pipeline: Pipeline = resposta_pipeline(context = self._texto, question = self._pergunta) | |
self._resposta = str(resposta_pipeline["answer"]) |