georad commited on
Commit
68eb640
·
verified ·
1 Parent(s): d1594e0

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. 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, fastest: all-MiniLM-L6-v2': 'all-MiniLM-L6-v2',
111
- 'fine-tuned model for medical domain: all-MiniLM-L6-v2': 'all-MiniLM-L6-v2',
112
- 'original model for general domain, best performance: all-mpnet-base-v2': 'all-mpnet-base-v2',
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, faster: meta-llama/Llama-3.2-1B-Instruct': 'meta-llama/Llama-3.2-1B-Instruct',
133
- 'fine-tuned model for medical domain: meta-llama/Llama-3.2-1B-Instruct': 'meta-llama/Llama-3.2-1B-Instruct',
134
- 'original model for general domain, slower: Qwen/Qwen2-1.5B-Instruct': 'Qwen/Qwen2-1.5B-Instruct',
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