Spaces:
Runtime error
Runtime error
taskswithcode
commited on
Commit
·
9cac99a
1
Parent(s):
93c89d0
Fix
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ import pdb
|
|
| 7 |
import json
|
| 8 |
from twc_embeddings import HFModel,SimCSEModel,SGPTModel,CausalLMModel,SGPTQnAModel
|
| 9 |
import torch
|
|
|
|
| 10 |
|
| 11 |
|
| 12 |
MAX_INPUT = 100
|
|
@@ -24,27 +25,36 @@ use_case_url = {"1":"https://huggingface.co/spaces/taskswithcode/semantic_simila
|
|
| 24 |
from transformers import BertTokenizer, BertForMaskedLM
|
| 25 |
|
| 26 |
|
|
|
|
|
|
|
| 27 |
|
| 28 |
-
view_count_file = "view_count.txt"
|
| 29 |
|
| 30 |
-
|
|
|
|
|
|
|
|
|
|
| 31 |
ret_val = 0
|
| 32 |
if ("view_count" not in st.session_state):
|
| 33 |
try:
|
| 34 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
except:
|
| 36 |
data = 0
|
| 37 |
-
data += 1
|
| 38 |
ret_val = data
|
| 39 |
st.session_state["view_count"] = data
|
| 40 |
-
with open(view_count_file,"w") as fp:
|
| 41 |
-
fp.write(str(data))
|
| 42 |
else:
|
| 43 |
ret_val = st.session_state["view_count"]
|
|
|
|
|
|
|
|
|
|
| 44 |
return "{:,}".format(ret_val)
|
| 45 |
|
| 46 |
|
| 47 |
|
|
|
|
| 48 |
def construct_model_info_for_display(model_names):
|
| 49 |
options_arr = []
|
| 50 |
markdown_str = f"<div style=\"font-size:16px; color: #2f2f2f; text-align: left\"><br/><b>Models evaluated ({len(model_names)})</b></div>"
|
|
@@ -154,6 +164,7 @@ def display_results(orig_sentences,main_index,results,response_info,app_mode):
|
|
| 154 |
main_sent = main_sent + "\n" + '\n'.join(body_sent)
|
| 155 |
st.markdown(main_sent,unsafe_allow_html=True)
|
| 156 |
st.session_state["download_ready"] = json.dumps(download_data,indent=4)
|
|
|
|
| 157 |
|
| 158 |
|
| 159 |
def init_session():
|
|
@@ -171,7 +182,7 @@ def app_main(app_mode,example_files,model_name_files):
|
|
| 171 |
curr_use_case = use_case[app_mode].split(".")[0]
|
| 172 |
st.markdown("<h5 style='text-align: center;'>Compare popular/state-of-the-art models for tasks using sentence embeddings</h5>", unsafe_allow_html=True)
|
| 173 |
st.markdown(f"<div style='color: #4f4f4f; text-align: left'>Use cases for sentence embeddings<br/> • <a href=\'{use_case_url['1']}\' target='_blank'>{use_case['1']}</a><br/> • {use_case['2']}<br/> • {use_case['3']}<br/><i>This app illustrates <b>'{curr_use_case}'</b> use case</i></div>", unsafe_allow_html=True)
|
| 174 |
-
st.markdown(f"<div style='color: #9f9f9f; text-align: right'>views: {get_views()}</div>", unsafe_allow_html=True)
|
| 175 |
|
| 176 |
|
| 177 |
try:
|
|
|
|
| 7 |
import json
|
| 8 |
from twc_embeddings import HFModel,SimCSEModel,SGPTModel,CausalLMModel,SGPTQnAModel
|
| 9 |
import torch
|
| 10 |
+
import requests
|
| 11 |
|
| 12 |
|
| 13 |
MAX_INPUT = 100
|
|
|
|
| 25 |
from transformers import BertTokenizer, BertForMaskedLM
|
| 26 |
|
| 27 |
|
| 28 |
+
APP_NAME = "hf/semantic_search"
|
| 29 |
+
INFO_URL = "http://www.taskswithcode.com/stats/"
|
| 30 |
|
|
|
|
| 31 |
|
| 32 |
+
|
| 33 |
+
|
| 34 |
+
def get_views(action):
|
| 35 |
+
print("in get views",action)
|
| 36 |
ret_val = 0
|
| 37 |
if ("view_count" not in st.session_state):
|
| 38 |
try:
|
| 39 |
+
print("inside get views")
|
| 40 |
+
app_info = {'name': APP_NAME,"action":action}
|
| 41 |
+
res = requests.post(INFO_URL, json = app_info).json()
|
| 42 |
+
print(res)
|
| 43 |
+
data = res["count"]
|
| 44 |
except:
|
| 45 |
data = 0
|
|
|
|
| 46 |
ret_val = data
|
| 47 |
st.session_state["view_count"] = data
|
|
|
|
|
|
|
| 48 |
else:
|
| 49 |
ret_val = st.session_state["view_count"]
|
| 50 |
+
if (action != "init"):
|
| 51 |
+
app_info = {'name': APP_NAME,"action":action}
|
| 52 |
+
res = requests.post(INFO_URL, json = app_info).json()
|
| 53 |
return "{:,}".format(ret_val)
|
| 54 |
|
| 55 |
|
| 56 |
|
| 57 |
+
|
| 58 |
def construct_model_info_for_display(model_names):
|
| 59 |
options_arr = []
|
| 60 |
markdown_str = f"<div style=\"font-size:16px; color: #2f2f2f; text-align: left\"><br/><b>Models evaluated ({len(model_names)})</b></div>"
|
|
|
|
| 164 |
main_sent = main_sent + "\n" + '\n'.join(body_sent)
|
| 165 |
st.markdown(main_sent,unsafe_allow_html=True)
|
| 166 |
st.session_state["download_ready"] = json.dumps(download_data,indent=4)
|
| 167 |
+
get_views("submit")
|
| 168 |
|
| 169 |
|
| 170 |
def init_session():
|
|
|
|
| 182 |
curr_use_case = use_case[app_mode].split(".")[0]
|
| 183 |
st.markdown("<h5 style='text-align: center;'>Compare popular/state-of-the-art models for tasks using sentence embeddings</h5>", unsafe_allow_html=True)
|
| 184 |
st.markdown(f"<div style='color: #4f4f4f; text-align: left'>Use cases for sentence embeddings<br/> • <a href=\'{use_case_url['1']}\' target='_blank'>{use_case['1']}</a><br/> • {use_case['2']}<br/> • {use_case['3']}<br/><i>This app illustrates <b>'{curr_use_case}'</b> use case</i></div>", unsafe_allow_html=True)
|
| 185 |
+
st.markdown(f"<div style='color: #9f9f9f; text-align: right'>views: {get_views('init')}</div>", unsafe_allow_html=True)
|
| 186 |
|
| 187 |
|
| 188 |
try:
|