Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import pipeline
 app = FastAPI()
 
 # Uncomment and set up the model if needed
-# model = pipeline('question-answering', model='bert-large-uncased-whole-word-masking-finetuned-squad', revision='main')
+model = pipeline('question-answering', model='bert-large-uncased-whole-word-masking-finetuned-squad', revision='main')
 
 # Serve static files (e.g., CSS, JS, images)
 app.mount("/static", StaticFiles(directory="static"), name="static")
@@ -21,6 +21,6 @@ async def home(request: Request):
 async def predict(request: Request):
     data = await request.json()
     user_input = data.get('message')
-    # result = model(user_input)
-    # response_text = result[0]['generated_text']
+    result = model(user_input)
+    response_text = result[0]['generated_text']
     return JSONResponse(content={'response': response_text})