Update app.py

app.py

@@ -30,6 +30,42 @@ asr_pipeline = pipeline(
     feature_extractor=processor.feature_extractor,  # Set the feature extractor for audio input
     device=device
 )
+# Load question-answering model
+model_name = 'pierreguillou/bert-base-cased-squad-v1.1-portuguese'
+qa_pipeline = pipeline("question-answering", model=model_name)
+
+# Define the context
+context = r"""
+A pandemia de COVID-19, também conhecida como pandemia de coronavírus, é uma pandemia em curso de COVID-19, 
+uma doença respiratória aguda causada pelo coronavírus da síndrome respiratória aguda grave 2 (SARS-CoV-2). 
+A doença foi identificada pela primeira vez em Wuhan, na província de Hubei, República Popular da China, 
+em 1 de dezembro de 2019, mas o primeiro caso foi reportado em 31 de dezembro do mesmo ano. 
+Acredita-se que o vírus tenha uma origem zoonótica, porque os primeiros casos confirmados 
+tinham principalmente ligações ao Mercado Atacadista de Frutos do Mar de Huanan, que também vendia animais vivos. 
+Em 11 de março de 2020, a Organização Mundial da Saúde declarou o surto uma pandemia. Até 8 de fevereiro de 2021, 
+pelo menos 105 743 102 casos da doença foram confirmados em pelo menos 191 países e territórios, 
+com cerca de 2 308 943 mortes e 58 851 440 pessoas curadas.
+"""
+
+# Define the request body for the POST method
+class QuestionRequest(BaseModel):
+    question: str
+
+# POST endpoint to answer questions
+@app.post("/answer/")
+async def answer_question(request: QuestionRequest):
+    try:
+        # Use the QA model to answer the question based on the context
+        result = qa_pipeline(question=request.question, context=context)
+        return {
+            "question": request.question,
+            "answer": result['answer'],
+            "score": round(result['score'], 4),
+            "start": result['start'],
+            "end": result['end']
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
 # Basic GET endpoint
 @app.get("/")