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:
|