taskswithcode commited on
Commit
44fc84f
1 Parent(s): 4ab3b97

Minor additions

Browse files
Files changed (1) hide show
  1. app.py +15 -2
app.py CHANGED
@@ -63,6 +63,18 @@ model_names = [
63
  "paper_url":"https://arxiv.org/abs/1908.10084",
64
  "mark":True,
65
  "class":"HFModel"},
 
 
 
 
 
 
 
 
 
 
 
 
66
 
67
  { "name":"SGPT-125M",
68
  "model":"Muennighoff/SGPT-125M-weightedmean-nli-bitfit",
@@ -148,7 +160,7 @@ example_file_names = {
148
 
149
  def construct_model_info_for_display():
150
  options_arr = []
151
- markdown_str = "<div style=\"font-size:16px; color: #2f2f2f; text-align: left\"><br/><b>Models evaluated</b></div>"
152
  for node in model_names:
153
  options_arr .append(node["name"])
154
  if (node["mark"] == True):
@@ -272,7 +284,8 @@ def main():
272
  options = list(dict.keys(example_file_names)), index=0, key = "twc_file")
273
  st.write("")
274
  options_arr,markdown_str = construct_model_info_for_display()
275
- selected_model = st.selectbox(label='Step 2. Select Model',
 
276
  options = options_arr, index=0, key = "twc_model")
277
  st.write("")
278
  main_index = st.number_input('Step 3. Enter index of sentence in file to make it the main sentence:',value=1,min_value = 1)
 
63
  "paper_url":"https://arxiv.org/abs/1908.10084",
64
  "mark":True,
65
  "class":"HFModel"},
66
+ { "name":"sentence-transformers/all-MiniLM-L12-v2",
67
+ "model":"sentence-transformers/all-MiniLM-L12-v2",
68
+ "fork_url":"https://github.com/taskswithcode/sentence_similarity_hf_model",
69
+ "orig_author_url":"https://github.com/UKPLab",
70
+ "orig_author":"Ubiquitous Knowledge Processing Lab",
71
+ "sota_info": {
72
+ "task":"Over 500,000 downloads from huggingface",
73
+ "sota_link":"https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2"
74
+ },
75
+ "paper_url":"https://arxiv.org/abs/1908.10084",
76
+ "mark":True,
77
+ "class":"HFModel"},
78
 
79
  { "name":"SGPT-125M",
80
  "model":"Muennighoff/SGPT-125M-weightedmean-nli-bitfit",
 
160
 
161
  def construct_model_info_for_display():
162
  options_arr = []
163
+ markdown_str = f"<div style=\"font-size:16px; color: #2f2f2f; text-align: left\"><br/><b>Models evaluated ({len(model_names)})</b></div>"
164
  for node in model_names:
165
  options_arr .append(node["name"])
166
  if (node["mark"] == True):
 
284
  options = list(dict.keys(example_file_names)), index=0, key = "twc_file")
285
  st.write("")
286
  options_arr,markdown_str = construct_model_info_for_display()
287
+ selection_label = 'Step 2. Select Model'
288
+ selected_model = st.selectbox(label=selection_label,
289
  options = options_arr, index=0, key = "twc_model")
290
  st.write("")
291
  main_index = st.number_input('Step 3. Enter index of sentence in file to make it the main sentence:',value=1,min_value = 1)