Update edubot.py

edubot.py

@@ -4,7 +4,7 @@ from langchain.vectorstores import FAISS
 from langchain.llms import CTransformers
 from langchain.chains import RetrievalQA
 from config import *
-import streamlit as st
+
 class EduBotCreator:
 
     def __init__(self):
@@ -23,7 +23,7 @@ class EduBotCreator:
         custom_prompt_temp = PromptTemplate(template=self.prompt_temp,
                             input_variables=self.input_variables)
         return custom_prompt_temp
-    @st.cache_resource()
+    
     def load_llm(self):
         llm = CTransformers(
                 model = self.model_ckpt,
@@ -32,7 +32,7 @@ class EduBotCreator:
                 temperature = self.temperature
             )
         return llm
-    @st.cache_resource()
+    
     def load_vectordb(self):
         hfembeddings = HuggingFaceEmbeddings(
                             model_name=self.embedder, 
@@ -42,7 +42,6 @@ class EduBotCreator:
         vector_db = FAISS.load_local(self.vector_db_path, hfembeddings)
         return vector_db
         
-    
     def create_bot(self, custom_prompt, vectordb, llm):
         retrieval_qa_chain = RetrievalQA.from_chain_type(
                                 llm=llm,