Update pages/type_text.py
Browse files- 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('
|
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('
|
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,
|