Update pages/type_text.py
Browse files- pages/type_text.py +8 -8
pages/type_text.py
CHANGED
@@ -107,10 +107,10 @@ numMAPPINGS_input = 5
|
|
107 |
|
108 |
## Define the Sentence Transformer models
|
109 |
st_models = {
|
110 |
-
'original model for general domain
|
111 |
-
'
|
112 |
-
'
|
113 |
-
'fine-tuned model for medical domain: all-mpnet-base-v2': 'all-mpnet-base-v2',
|
114 |
}
|
115 |
|
116 |
#model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
|
@@ -129,10 +129,10 @@ SentTrans_model = st_models[selected_st_model]
|
|
129 |
|
130 |
## Define the Reasoning models
|
131 |
rs_models = {
|
132 |
-
'original model for general domain
|
133 |
-
'
|
134 |
-
'
|
135 |
-
'fine-tuned model for medical domain: Qwen/Qwen2-1.5B-Instruct': 'Qwen/Qwen2-1.5B-Instruct',
|
136 |
}
|
137 |
|
138 |
## Create the select Reasoning box
|
|
|
107 |
|
108 |
## Define the Sentence Transformer models
|
109 |
st_models = {
|
110 |
+
'(fastest performance) original model for general domain: all-MiniLM-L6-v2': 'all-MiniLM-L6-v2',
|
111 |
+
'(high performance) original model for general domain: all-mpnet-base-v2': 'all-mpnet-base-v2',
|
112 |
+
'(expected in future) fine-tuned model for medical domain: all-MiniLM-L6-v2': 'all-MiniLM-L6-v2',
|
113 |
+
'(expected in future) fine-tuned model for medical domain: all-mpnet-base-v2': 'all-mpnet-base-v2',
|
114 |
}
|
115 |
|
116 |
#model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
|
|
|
129 |
|
130 |
## Define the Reasoning models
|
131 |
rs_models = {
|
132 |
+
'(faster performance) original model for general domain: meta-llama/Llama-3.2-1B-Instruct': 'meta-llama/Llama-3.2-1B-Instruct',
|
133 |
+
'(slower performance) original model for general domain: Qwen/Qwen2-1.5B-Instruct': 'Qwen/Qwen2-1.5B-Instruct',
|
134 |
+
'(expected in future) fine-tuned model for medical domain: meta-llama/Llama-3.2-1B-Instruct': 'meta-llama/Llama-3.2-1B-Instruct',
|
135 |
+
'(expected in future) fine-tuned model for medical domain: Qwen/Qwen2-1.5B-Instruct': 'Qwen/Qwen2-1.5B-Instruct',
|
136 |
}
|
137 |
|
138 |
## Create the select Reasoning box
|