Update app.py

app.py

@@ -16,7 +16,7 @@ from langchain.vectorstores import Chroma
 
 
 @st.cache_resource
-def load_model(model_name, logger, ):
+def load_model(model_name):
     logger.info("Loading model ..")
     start_time = time.time()
 
@@ -49,7 +49,7 @@ def load_model(model_name, logger, ):
 
 
 @st.cache_resource
-def load_db(logger, device, local_embed=False,  CHROMA_PATH = './ChromaDB'):
+def load_db(device, local_embed=False,  CHROMA_PATH = './ChromaDB'):
     """
     Load vector embeddings and Chroma database 
     """
@@ -278,9 +278,9 @@ if __name__=="__main__":
                 Question: {question}\n> Context:\n>>>\n{context}\n>>>\nRelevant parts"""}
 
 
-    db = load_db(logger, device)
+    db = load_db(device)
 
-    model, tokenizer = load_model(model_name, logger)
+    model, tokenizer = load_model(model_name)
     
     # streamlit chat
     if user_question is not None and user_question!="":