Spaces:
Runtime error
Runtime error
Commit
·
18b4b80
1
Parent(s):
b1ecb1f
Update app.py
Browse files
app.py
CHANGED
|
@@ -53,10 +53,13 @@ def get_vectorstore(text_chunks):
|
|
| 53 |
vectorstore = FAISS.from_texts(texts=text_chunks, embedding=embeddings)
|
| 54 |
return vectorstore
|
| 55 |
|
|
|
|
|
|
|
|
|
|
| 56 |
def get_conversation_chain(vectorstore):
|
| 57 |
#llm = OpenAI()
|
| 58 |
#llm = ChatOpenAI()
|
| 59 |
-
llm = HuggingFaceHub(repo_id="
|
| 60 |
memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True)
|
| 61 |
conversation_chain = ConversationalRetrievalChain.from_llm(
|
| 62 |
llm=llm,
|
|
|
|
| 53 |
vectorstore = FAISS.from_texts(texts=text_chunks, embedding=embeddings)
|
| 54 |
return vectorstore
|
| 55 |
|
| 56 |
+
|
| 57 |
+
# google/flan-t5-xxl
|
| 58 |
+
|
| 59 |
def get_conversation_chain(vectorstore):
|
| 60 |
#llm = OpenAI()
|
| 61 |
#llm = ChatOpenAI()
|
| 62 |
+
llm = HuggingFaceHub(repo_id="bigscience/bloom", model_kwargs={"temperature":0.5, "max_length":512})
|
| 63 |
memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True)
|
| 64 |
conversation_chain = ConversationalRetrievalChain.from_llm(
|
| 65 |
llm=llm,
|