Spaces:
Build error
Build error
Mya-Mya
commited on
Commit
·
4db15a6
1
Parent(s):
e9b49df
Delete localtest.py
Browse files- localtest.py +0 -35
localtest.py
DELETED
@@ -1,35 +0,0 @@
|
|
1 |
-
from backend import Backend
|
2 |
-
import frontend
|
3 |
-
import numpy as np
|
4 |
-
from pandas import DataFrame
|
5 |
-
import pickle
|
6 |
-
|
7 |
-
with open("./himitsudogu_db.pkl", "rb") as file:
|
8 |
-
himitsudogu_db: dict = pickle.load(file)
|
9 |
-
|
10 |
-
class LocaltestBackend(Backend):
|
11 |
-
"""
|
12 |
-
ローカル開発環境にSentence BERTを連れ込んでくるのは無理なので、
|
13 |
-
文章をベクトル化する行程は乱数発生とする。
|
14 |
-
"""
|
15 |
-
|
16 |
-
def __init__(self):
|
17 |
-
super().__init__()
|
18 |
-
self.feature_matrix = himitsudogu_db["feature_matrix_s"][
|
19 |
-
"sonoisa/sentence-bert-base-ja-mean-tokens-v2"
|
20 |
-
]
|
21 |
-
|
22 |
-
def on_submit_button_press(self, query: str) -> DataFrame:
|
23 |
-
query_feature_vector = np.random.random((768,))
|
24 |
-
cs_s = self.feature_matrix @ query_feature_vector
|
25 |
-
ranked_index_s = np.argsort(cs_s)[::-1]
|
26 |
-
output = DataFrame(columns=["類似度", "名前", "説明"])
|
27 |
-
for rank, i in enumerate(ranked_index_s[:20], 1):
|
28 |
-
output.loc[rank] = [
|
29 |
-
cs_s[i],
|
30 |
-
himitsudogu_db["name_s"][i],
|
31 |
-
himitsudogu_db["description_s"][i],
|
32 |
-
]
|
33 |
-
return output
|
34 |
-
|
35 |
-
frontend.launch_frontend(backend=LocaltestBackend())
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|