sonoisa commited on
Commit
33ac704
1 Parent(s): 67236f5

Update model to v1

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -146,14 +146,14 @@ description_text = st.empty()
146
 
147
  if "model" not in st.session_state:
148
  description_text.text("...モデル読み込み中...")
149
- model = SentenceBertJapanese("sonoisa/sentence-bert-base-ja-mean-tokens-v2")
150
  st.session_state.model = model
151
 
152
  pyminizip.uncompress(
153
- "irasuto_items_20210224_v2.pq.zip", st.secrets["ZIP_PASSWORD"], None, 1
154
  )
155
 
156
- df = pq.read_table("irasuto_items_20210224_v2.parquet").to_pandas()
157
  st.session_state.df = df
158
 
159
  sentence_vectors = np.stack(df["sentence_vector"])
 
146
 
147
  if "model" not in st.session_state:
148
  description_text.text("...モデル読み込み中...")
149
+ model = SentenceBertJapanese("sonoisa/sentence-bert-base-ja-mean-tokens")
150
  st.session_state.model = model
151
 
152
  pyminizip.uncompress(
153
+ "irasuto_items_20210224.pq.zip", st.secrets["ZIP_PASSWORD"], None, 1
154
  )
155
 
156
+ df = pq.read_table("irasuto_items_20210224.parquet").to_pandas()
157
  st.session_state.df = df
158
 
159
  sentence_vectors = np.stack(df["sentence_vector"])