File size: 651 Bytes
85784d8 82c5428 85784d8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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()
@property
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"]) |