Update app.py

app.py

@@ -123,7 +123,7 @@ def ask_question(question):
                     "context": "..." + answer["context"] + "...",
                     "answer": answer["answer"],
                     "relevance": round(answer["score"] * 100, 2),
-                    "document": [doc for doc in response["documents"] if doc["id"] == answer["document_id"]][0],
+                    "document": [doc for doc in prediction["documents"] if doc["id"] == answer["document_id"]][0],
                     # "_raw": answer,
                     "offset_start_in_doc": answer["offsets_in_document"][0]["start"],
                 }