add bert

app.py

@@ -6,6 +6,7 @@ from fastapi import FastAPI, UploadFile, File
 from typing import Union 
 import json 
 import csv
+from modeles import bert
 
 
 app = FastAPI()
@@ -46,6 +47,13 @@ async def create_upload_file(texte: str, model: str):
 
     return {"model": model, "texte": texte}
 
+@app.post("/bert/")
+async def qabert(context: str, question: str):
+    bertAnswer = bert(context, question)
+    if bert:
+        return bert
+    else:
+        raise HTTPException(status_code=400, detail="Error")
 
 def extract_data(file: UploadFile) -> Union[str, dict, list]:
     if file.filename.endswith(".txt"):

modeles.py

@@ -0,0 +1,5 @@
+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))