boryasbora commited on
Commit
cea573b
·
verified ·
1 Parent(s): 386e794

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -14,7 +14,7 @@ from langchain_core.output_parsers import StrOutputParser
14
  from langchain_core.runnables import RunnableLambda
15
  from datetime import date
16
  import time
17
- llm_list = ['Mistral-7B-Instruct-v0.2','Mixtral-8x7B-Instruct-v0.1','LLAMA3']
18
  blablador_base = "https://helmholtz-blablador.fz-juelich.de:8000/v1"
19
  # Environment variables
20
  os.environ['LANGCHAIN_TRACING_V2'] = 'true'
@@ -122,13 +122,11 @@ with st.sidebar:
122
  index = None)
123
 
124
  temperature = st.slider("Temperature: ", 0.0, 1.0, 0.5, 0.1)
125
- if selected_model in ['Mistral-7B-Instruct-v0.2', 'Mixtral-8x7B-Instruct-v0.1','LLAMA3']:
126
  if selected_model == 'Mistral-7B-Instruct-v0.2':
127
  selected_model = 'alias-fast'
128
  elif selected_model == 'Mixtral-8x7B-Instruct-v0.1':
129
  selected_model = 'alias-large'
130
- elif selected_model == 'LLAMA3':
131
- selected_model = 'alias-experimental'
132
  chain = get_chain(temperature,selected_model)
133
  st.button('Clear Chat History', on_click=clear_chat_history)
134
 
 
14
  from langchain_core.runnables import RunnableLambda
15
  from datetime import date
16
  import time
17
+ llm_list = ['Mistral-7B-Instruct-v0.2','Mixtral-8x7B-Instruct-v0.1']
18
  blablador_base = "https://helmholtz-blablador.fz-juelich.de:8000/v1"
19
  # Environment variables
20
  os.environ['LANGCHAIN_TRACING_V2'] = 'true'
 
122
  index = None)
123
 
124
  temperature = st.slider("Temperature: ", 0.0, 1.0, 0.5, 0.1)
125
+ if selected_model in ['Mistral-7B-Instruct-v0.2', 'Mixtral-8x7B-Instruct-v0.1']:
126
  if selected_model == 'Mistral-7B-Instruct-v0.2':
127
  selected_model = 'alias-fast'
128
  elif selected_model == 'Mixtral-8x7B-Instruct-v0.1':
129
  selected_model = 'alias-large'
 
 
130
  chain = get_chain(temperature,selected_model)
131
  st.button('Clear Chat History', on_click=clear_chat_history)
132