Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ from langchain.vectorstores import Chroma
|
|
| 9 |
from langchain.embeddings import HuggingFaceEmbeddings
|
| 10 |
from langchain.document_loaders import JSONLoader, DirectoryLoader
|
| 11 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
| 12 |
-
from
|
| 13 |
from langchain.chains import RetrievalQA
|
| 14 |
from langchain.prompts import PromptTemplate
|
| 15 |
from langchain.schema import Document
|
|
@@ -55,7 +55,7 @@ class LangChainRAGSystem:
|
|
| 55 |
)
|
| 56 |
|
| 57 |
# Initialize LLM
|
| 58 |
-
self.llm =
|
| 59 |
groq_api_key=GROQ_API_KEY,
|
| 60 |
model_name=GROQ_MODEL,
|
| 61 |
temperature=0.1,
|
|
|
|
| 9 |
from langchain.embeddings import HuggingFaceEmbeddings
|
| 10 |
from langchain.document_loaders import JSONLoader, DirectoryLoader
|
| 11 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
| 12 |
+
from langchain_groq import ChatGroq
|
| 13 |
from langchain.chains import RetrievalQA
|
| 14 |
from langchain.prompts import PromptTemplate
|
| 15 |
from langchain.schema import Document
|
|
|
|
| 55 |
)
|
| 56 |
|
| 57 |
# Initialize LLM
|
| 58 |
+
self.llm = ChatGroq(
|
| 59 |
groq_api_key=GROQ_API_KEY,
|
| 60 |
model_name=GROQ_MODEL,
|
| 61 |
temperature=0.1,
|