got rid of json

app.py

@@ -4,11 +4,11 @@ import numpy as np
 import faiss
 import PyPDF2
 import os
+import langchain
 
 from transformers import DPRContextEncoder, DPRContextEncoderTokenizer, DPRQuestionEncoder, DPRQuestionEncoderTokenizer, BartForQuestionAnswering
 from transformers import BartForConditionalGeneration, BartTokenizer, AutoTokenizer
 
-from langchain import text_splitter
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.document_loaders import PyPDFLoader
 
@@ -151,6 +151,10 @@ question_model_name="facebook/dpr-question_encoder-multiset-base"
 
 rag = RAG(file_path, device)
 
+query = "what is the benefit of using multiple attention heads in mult-head attention?"
+
+print(rag.query(query))
+
 st.title("RAG Model Query Interface")
 
 query = st.text_area("Enter your question:")
@@ -158,4 +162,4 @@ query = st.text_area("Enter your question:")
 # If a query is given, get the answer
 if query:
     answer = rag.query(query)
-    st.json(answer)
+    st.write(answer)