georad commited on
Commit
ebcace0
·
verified ·
1 Parent(s): dd077c3

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +3 -3
pages/type_text.py CHANGED
@@ -133,7 +133,7 @@ st_models = {
133
  #model = SentenceTransformer('clips/mfaq')
134
 
135
  ## Create the select Sentence Transformer box
136
- selected_st_model = st.selectbox('Current selected Sentence Transformer model:', list(st_models.keys())) # or 'Choose a Sentence Transformer Model'
137
  #st.write("Current selection:", selected_st_model)
138
 
139
  ## Get the selected SentTrans model
@@ -184,7 +184,7 @@ if INTdesc_input and st.button(":blue[Map to SBS codes]", key="run_st_model"): #
184
  }
185
 
186
  ## Create the select Reasoning box
187
- selected_rs_model = st.selectbox('Current selected Reasoning model:', list(rs_models.keys())) # or 'Choose a Reasoning Model'
188
  #st.write("Current selection:", selected_rs_model)
189
 
190
  ## Get the selected Reasoning model
@@ -212,7 +212,7 @@ if INTdesc_input and st.button(":blue[Map to SBS codes]", key="run_st_model"): #
212
 
213
  st.write("") # optional Add some space before the Reasoning Model output
214
  st.subheader("Reasoning Model Output:") # Added a subheader for clarity
215
- with st.spinner("Running Reasoning Model... may take several minutes, grab a coffee"): # Added spinner for Reasoning model
216
  outputs = pipe(
217
  messages,
218
  max_new_tokens=256,
 
133
  #model = SentenceTransformer('clips/mfaq')
134
 
135
  ## Create the select Sentence Transformer box
136
+ selected_st_model = st.selectbox('Sentence Transformer model:', list(st_models.keys())) # or 'Choose a Sentence Transformer Model'
137
  #st.write("Current selection:", selected_st_model)
138
 
139
  ## Get the selected SentTrans model
 
184
  }
185
 
186
  ## Create the select Reasoning box
187
+ selected_rs_model = st.selectbox('Reasoning model:', list(rs_models.keys())) # or 'Choose a Reasoning Model'
188
  #st.write("Current selection:", selected_rs_model)
189
 
190
  ## Get the selected Reasoning model
 
212
 
213
  st.write("") # optional Add some space before the Reasoning Model output
214
  st.subheader("Reasoning Model Output:") # Added a subheader for clarity
215
+ with st.spinner("Running Reasoning Model... may take several minutes, grab a :coffee:"): # Added spinner for Reasoning model
216
  outputs = pipe(
217
  messages,
218
  max_new_tokens=256,