Spaces:

tien314
/

hscode_classification

Running

tien314 commited on Oct 11, 2024

Commit

ccf7fbf

verified ·

1 Parent(s): 7b04950

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ def load_data():
     corpus = [doc for doc in df['document'].to_list()]
     retriever = bm25s.BM25(corpus=corpus)
-    retriever.index(bm25s.tokenize(corpus)
     return retriever

     corpus = [doc for doc in df['document'].to_list()]
     retriever = bm25s.BM25(corpus=corpus)
+    retriever.index(bm25s.tokenize(corpus))
     return retriever