Update app.py

app.py

@@ -165,11 +165,12 @@ def loadKB(fileprovided, urlProvided, uploads_dir, request):
     return documents
 
 
-def getRAGChain(customerName,customerDistrict, custDetailsPresent,vectordb):
+def getRAGChain(customerName, customerDistrict, custDetailsPresent, vectordb,llmID):
     chain = RetrievalQA.from_chain_type(
-        llm=getLLMModel(0),
+        llm=getLLMModel(llmID),
         chain_type='stuff',
-        retriever=vectordb.as_retriever(),
+        retriever=getRetriever(vectordb),
+        #retriever=vectordb.as_retriever(),
         verbose=False,
         chain_type_kwargs={
             "verbose": False,