modify - bert

app.py

@@ -48,12 +48,12 @@ async def create_upload_file(texte: str, model: str):
     return {"model": model, "texte": texte}
 
 @app.post("/bert/")
-async def qabert(context: str, question: str):
+async def qabert(context, question):
     bertAnswer = bert(context, question)
-    if bert:
-        return bert
+    if bertAnswer:
+        return bertAnswer
     else:
-        raise HTTPException(status_code=400, detail="Error")
+        raise HTTPException(status_code=400, detail="Error while asking to bert")
 
 def extract_data(file: UploadFile) -> Union[str, dict, list]:
     if file.filename.endswith(".txt"):