Update app.py
Browse files
app.py
CHANGED
|
@@ -20,6 +20,8 @@ genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
|
|
| 20 |
|
| 21 |
llm = genai.GenerativeModel('gemini-pro')
|
| 22 |
|
|
|
|
|
|
|
| 23 |
# Streamlit UI
|
| 24 |
st.title("AI Response Generator")
|
| 25 |
|
|
@@ -32,7 +34,6 @@ documents = loader.load()
|
|
| 32 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
|
| 33 |
texts = text_splitter.split_documents(documents)
|
| 34 |
|
| 35 |
-
instructor_embeddings = HuggingFaceInstructEmbeddings(model_name="hkunlp/instructor-xl", model_kwargs={"device": "cuda"})
|
| 36 |
|
| 37 |
persist_directory = 'db'
|
| 38 |
embedding = instructor_embeddings
|
|
|
|
| 20 |
|
| 21 |
llm = genai.GenerativeModel('gemini-pro')
|
| 22 |
|
| 23 |
+
instructor_embeddings = HuggingFaceInstructEmbeddings(model_name="hkunlp/instructor-base", model_kwargs={"device": "cpu"})
|
| 24 |
+
|
| 25 |
# Streamlit UI
|
| 26 |
st.title("AI Response Generator")
|
| 27 |
|
|
|
|
| 34 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
|
| 35 |
texts = text_splitter.split_documents(documents)
|
| 36 |
|
|
|
|
| 37 |
|
| 38 |
persist_directory = 'db'
|
| 39 |
embedding = instructor_embeddings
|