Update app.py

app.py

@@ -132,7 +132,7 @@ def rag_workflow(query):
 
     
     # Step 2: Retrieve contexts (documents and code snippets)
-    kadiAPY_doc_documents, kadiAPY_code_documents = rag_chain.retrieve_contexts(query, library_usage_prediction)
+    kadiAPY_doc_documents, kadiAPY_code_documents = rag_chain.retrieve_contexts(query, code_library_usage_prediction)
     print("Retrieved Document Contexts:", doc_contexts)
     print("Retrieved Code Contexts:", code_contexts)