Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +44,7 @@ if st.session_state.retriever is None:
|
|
44 |
sentence = st.text_input("please enter description:")
|
45 |
|
46 |
if sentence !='':
|
47 |
-
results,_ = st.session_state.retriever.retrieve(bm25s.tokenize(
|
48 |
doc = [d for d in results]
|
49 |
hscodes = [extract_hscode(item) for item in doc[0]]
|
50 |
for code in hscodes:
|
|
|
44 |
sentence = st.text_input("please enter description:")
|
45 |
|
46 |
if sentence !='':
|
47 |
+
results,_ = st.session_state.retriever.retrieve(bm25s.tokenize(sentence), k=5)
|
48 |
doc = [d for d in results]
|
49 |
hscodes = [extract_hscode(item) for item in doc[0]]
|
50 |
for code in hscodes:
|