Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,6 @@ from langchain.memory import ConversationBufferMemory
|
|
17 |
from langchain.chains import ConversationalRetrievalChain
|
18 |
from htmlTemplates import css, bot_template, user_template
|
19 |
from langchain.llms import HuggingFaceHub
|
20 |
-
from huggingface_hub import InferenceApi
|
21 |
|
22 |
|
23 |
def get_pdf_text(pdf_docs):
|
@@ -106,8 +105,8 @@ def get_conversation_chain(vectorstore):
|
|
106 |
A conversational retrieval chain for generating responses.
|
107 |
|
108 |
"""
|
109 |
-
llm =
|
110 |
-
repo_id="mistralai/Mixtral-
|
111 |
model_kwargs={"temperature": 0.5, "max_length": 1048},
|
112 |
)
|
113 |
# llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613")
|
@@ -192,4 +191,4 @@ def main():
|
|
192 |
|
193 |
|
194 |
if __name__ == "__main__":
|
195 |
-
main()
|
|
|
17 |
from langchain.chains import ConversationalRetrievalChain
|
18 |
from htmlTemplates import css, bot_template, user_template
|
19 |
from langchain.llms import HuggingFaceHub
|
|
|
20 |
|
21 |
|
22 |
def get_pdf_text(pdf_docs):
|
|
|
105 |
A conversational retrieval chain for generating responses.
|
106 |
|
107 |
"""
|
108 |
+
llm = HuggingFaceHub(
|
109 |
+
repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
|
110 |
model_kwargs={"temperature": 0.5, "max_length": 1048},
|
111 |
)
|
112 |
# llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613")
|
|
|
191 |
|
192 |
|
193 |
if __name__ == "__main__":
|
194 |
+
main()
|