aloqas-qa-fastapi / modeles.py
alexandre-huynh's picture
Update modeles.py
9d71420 verified
raw
history blame
350 Bytes
# transformers obtenu via pip install automatique spécifié dans requirements.txt
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))