SiraH commited on
Commit
addae57
1 Parent(s): 6fbbd3e

import tempfile library

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,6 +3,7 @@ import streamlit as st
3
  from langchain.text_splitter import RecursiveCharacterTextSplitter
4
  import re
5
  import pathlib
 
6
 
7
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
8
  from langchain.llms import HuggingFacePipeline
@@ -239,8 +240,8 @@ def main():
239
  st.write(sp_docs)
240
  sp_docs_list.extend(sp_docs)
241
 
242
- st.write(sp_docs_list[0])
243
- db = FAISS.from_documents(sp_docs_list[0], embeddings)
244
  memory = ConversationBufferMemory(memory_key="chat_history",
245
  return_messages=True,
246
  input_key="query",
 
3
  from langchain.text_splitter import RecursiveCharacterTextSplitter
4
  import re
5
  import pathlib
6
+ from tempfile import NamedTemporaryFile
7
 
8
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
9
  from langchain.llms import HuggingFacePipeline
 
240
  st.write(sp_docs)
241
  sp_docs_list.extend(sp_docs)
242
 
243
+ st.write(sp_docs_list)
244
+ db = FAISS.from_documents(sp_docs_list, embeddings)
245
  memory = ConversationBufferMemory(memory_key="chat_history",
246
  return_messages=True,
247
  input_key="query",