Update app.py

app.py

@@ -20,6 +20,8 @@ genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
 
 llm = genai.GenerativeModel('gemini-pro')
 
+instructor_embeddings = HuggingFaceInstructEmbeddings(model_name="hkunlp/instructor-base", model_kwargs={"device": "cpu"})
+
 # Streamlit UI
 st.title("AI Response Generator")
 
@@ -32,7 +34,6 @@ documents = loader.load()
 text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
 texts = text_splitter.split_documents(documents)
 
-instructor_embeddings = HuggingFaceInstructEmbeddings(model_name="hkunlp/instructor-xl", model_kwargs={"device": "cuda"})
 
 persist_directory = 'db'
 embedding = instructor_embeddings