georad commited on
Commit
9f5f8bf
·
verified ·
1 Parent(s): 34d01ba

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +6 -6
pages/type_text.py CHANGED
@@ -99,7 +99,7 @@ numMAPPINGS_input = 5
99
 
100
  ## Define the Sentence Transformer models
101
  st_models = {
102
- '(fastest performance) original model for general domain: all-MiniLM-L6-v2': 'all-MiniLM-L6-v2',
103
  '(high performance) original model for general domain: all-mpnet-base-v2': 'all-mpnet-base-v2',
104
  '(expected in future) fine-tuned model for medical domain: all-MiniLM-L6-v2': 'all-MiniLM-L6-v2',
105
  '(expected in future) fine-tuned model for medical domain: all-mpnet-base-v2': 'all-mpnet-base-v2',
@@ -112,8 +112,8 @@ st_models = {
112
  #model = SentenceTransformer('clips/mfaq')
113
 
114
  ## Create the select Sentence Transformer box
115
- selected_st_model = st.selectbox('Choose a Sentence Transformer model:', list(st_models.keys()))
116
- st.write("Current selection:", selected_st_model)
117
 
118
  ## Get the selected SentTrans model
119
  SentTrans_model = st_models[selected_st_model]
@@ -121,14 +121,14 @@ SentTrans_model = st_models[selected_st_model]
121
 
122
  ## Define the Reasoning models
123
  rs_models = {
124
- '(faster performance) original model for general domain: meta-llama/Llama-3.2-1B-Instruct': 'meta-llama/Llama-3.2-1B-Instruct',
125
- '(slower performance) original model for general domain: Qwen/Qwen2-1.5B-Instruct': 'Qwen/Qwen2-1.5B-Instruct',
126
  '(expected in future) fine-tuned model for medical domain: meta-llama/Llama-3.2-1B-Instruct': 'meta-llama/Llama-3.2-1B-Instruct',
127
  '(expected in future) fine-tuned model for medical domain: Qwen/Qwen2-1.5B-Instruct': 'Qwen/Qwen2-1.5B-Instruct',
128
  }
129
 
130
  ## Create the select Reasoning box
131
- selected_rs_model = st.selectbox('Current selected Reasoning model:', list(rs_models.keys()))
132
  #st.write("Current selection:", selected_rs_model)
133
 
134
  ## Get the selected Reasoning model
 
99
 
100
  ## Define the Sentence Transformer models
101
  st_models = {
102
+ '(higher speed) original model for general domain: all-MiniLM-L6-v2': 'all-MiniLM-L6-v2',
103
  '(high performance) original model for general domain: all-mpnet-base-v2': 'all-mpnet-base-v2',
104
  '(expected in future) fine-tuned model for medical domain: all-MiniLM-L6-v2': 'all-MiniLM-L6-v2',
105
  '(expected in future) fine-tuned model for medical domain: all-mpnet-base-v2': 'all-mpnet-base-v2',
 
112
  #model = SentenceTransformer('clips/mfaq')
113
 
114
  ## Create the select Sentence Transformer box
115
+ selected_st_model = st.selectbox('Current selected Sentence Transformer model:', list(st_models.keys())) # or 'Choose a Sentence Transformer Model'
116
+ #st.write("Current selection:", selected_st_model)
117
 
118
  ## Get the selected SentTrans model
119
  SentTrans_model = st_models[selected_st_model]
 
121
 
122
  ## Define the Reasoning models
123
  rs_models = {
124
+ '(medium speed) original model for general domain: meta-llama/Llama-3.2-1B-Instruct': 'meta-llama/Llama-3.2-1B-Instruct',
125
+ '(slower speed) original model for general domain: Qwen/Qwen2-1.5B-Instruct': 'Qwen/Qwen2-1.5B-Instruct',
126
  '(expected in future) fine-tuned model for medical domain: meta-llama/Llama-3.2-1B-Instruct': 'meta-llama/Llama-3.2-1B-Instruct',
127
  '(expected in future) fine-tuned model for medical domain: Qwen/Qwen2-1.5B-Instruct': 'Qwen/Qwen2-1.5B-Instruct',
128
  }
129
 
130
  ## Create the select Reasoning box
131
+ selected_rs_model = st.selectbox('Current selected Reasoning model:', list(rs_models.keys())) # or 'Choose a Reasoning Model'
132
  #st.write("Current selection:", selected_rs_model)
133
 
134
  ## Get the selected Reasoning model