aloqas-qa-fastapi / modeles.py
LucasAguetai
add bert
b4d0d8e
raw
history blame
267 Bytes
from transformers import pipeline
def bert(context, question):
question_answerer = pipeline("question-answering", "alexandre-huynh/bert-base-uncased-finetuned-squad-single-epoch", framework="tf")
return(question_answerer(context=context, question=question))