Update app.py

app.py

@@ -248,7 +248,7 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
         else:
             if st.session_state.documents_only:
                 prompt = idk(query)
-                answer = gen_text(prompt, max_tokens=128, sampling_params=SamplingParams(temperature=0.6, top_p=0.9))
+                answer = json.loads(gen_text(prompt, max_tokens=128, sampling_params=SamplingParams(temperature=0.6, top_p=0.9)))['answer']
                 print(f'TYPE: {type(answer)}')
             else:
                 prompt = self_knowledge(query)
@@ -257,7 +257,7 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
 
     torch.cuda.empty_cache()
     
-    return json.loads(answer)['answer']
+    return answer
 
 def collect_files(conn, cursor, directory, pattern):
     array = []