sonoisa commited on
Commit
3faad04
1 Parent(s): e665ead

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -164,10 +164,10 @@ prev_query = ""
164
  query_input = st.text_input(label="説明文", value="")
165
  search_buttion = st.button("検索")
166
 
167
- closest_n = 10
168
 
169
- if search_buttion or prev_query != str(query_input):
170
- query = str(query_input)
171
  prev_query = query
172
  query_embedding = model.encode([query]).numpy()
173
 
 
164
  query_input = st.text_input(label="説明文", value="")
165
  search_buttion = st.button("検索")
166
 
167
+ closest_n = st.number_input(label="検索数", min_value=1, value=10, max_value=100)
168
 
169
+ if search_buttion or prev_query != query_input:
170
+ query = query_input
171
  prev_query = query
172
  query_embedding = model.encode([query]).numpy()
173