Spaces:
Runtime error
Runtime error
taskswithcode
commited on
Commit
•
44fc84f
1
Parent(s):
4ab3b97
Minor additions
Browse files
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 |
-
|
|
|
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)
|