Hackoor commited on
Commit
a51a087
·
1 Parent(s): 15d23f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -65,7 +65,7 @@ def create_conversational_chain(vector_store):
65
  #model_type="llama", config={'max_new_tokens': 500, 'temperature': 0.01})
66
  llm = Replicate(
67
  streaming = True,
68
- model = "replicate/llama-2-70b-chat:58d078176e02c219e11eb4da5a02a7830a283b14cf8f94537af893ccff5ee781",
69
  callbacks=[StreamingStdOutCallbackHandler()],
70
  input = {"temperature": 0.01, "max_length" :500,"top_p":1})
71
  memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
@@ -79,7 +79,7 @@ def main():
79
  load_dotenv()
80
  # Initialize session state
81
  initialize_session_state()
82
- st.title("Multi-Docs ChatBot using llama-2-70b :books:")
83
  # Initialize Streamlit
84
  st.sidebar.title("Document Processing")
85
  uploaded_files = st.sidebar.file_uploader("Upload files", accept_multiple_files=True)
 
65
  #model_type="llama", config={'max_new_tokens': 500, 'temperature': 0.01})
66
  llm = Replicate(
67
  streaming = True,
68
+ model = "NousResearch/Llama-2-7b-hf",
69
  callbacks=[StreamingStdOutCallbackHandler()],
70
  input = {"temperature": 0.01, "max_length" :500,"top_p":1})
71
  memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
 
79
  load_dotenv()
80
  # Initialize session state
81
  initialize_session_state()
82
+ st.title("Multi-Docs ChatBot using llama-2-7b :books:")
83
  # Initialize Streamlit
84
  st.sidebar.title("Document Processing")
85
  uploaded_files = st.sidebar.file_uploader("Upload files", accept_multiple_files=True)