jonathanjordan21 commited on
Commit
2e3f8d8
1 Parent(s): 675db3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ API_TOKEN = os.getenv('HF_INFER_API')
25
  @st.cache_resource
26
  def get_llm_chain():
27
  return custom_chain_with_history(
28
- llm=CustomLLM(repo_id="epfl-llm/meditron-70b", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"], temperature=0.001),
29
  # llm=CustomLLM(repo_id="google/gemma-7b", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"], temperature=0.001),
30
  # memory=st.session_state.memory.chat_memory,
31
  memory=st.session_state.memory
 
25
  @st.cache_resource
26
  def get_llm_chain():
27
  return custom_chain_with_history(
28
+ llm=CustomLLM(repo_id="AdaptLLM/medicine-LLM", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"], temperature=0.001),
29
  # llm=CustomLLM(repo_id="google/gemma-7b", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"], temperature=0.001),
30
  # memory=st.session_state.memory.chat_memory,
31
  memory=st.session_state.memory