update app

app.py

@@ -11,7 +11,7 @@ sys.path.append(str(path_root))
 encoder_index_map = {
     'uniCOIL': ('UniCoil', 'castorini/unicoil-noexp-msmarco-passage', 'index-unicoil'),
     'SPLADE++ Ensemble Distil': ('SpladePlusPlusEnsembleDistil', 'naver/splade-cocondenser-ensembledistil', 'index-splade-pp-ed'),
-    'SPLADE++ Self Distil': ('SpladePlusPlusSelfDistil', 'naver/splade-cocondenser-ensembledistil', 'index-splade-pp-sd')
+    'SPLADE++ Self Distil': ('SpladePlusPlusSelfDistil', 'naver/splade-cocondenser-selfdistil', 'index-splade-pp-sd')
 }
 
 index = 'index-splade-pp-ed'
@@ -65,7 +65,6 @@ with col2:
 if search_query or button_clicked:
     num_results = None
     t_0 = time.time()
-    print("search query is:\t", search_query)
     search_results = searcher.search(search_query, k=10)
     search_time = time.time() - t_0
     st.write(
@@ -73,7 +72,12 @@ if search_query or button_clicked:
     for i, result in enumerate(search_results[:10]):
         result_score = result.score
         result_id = result.docid
-        contents = json.loads(result.raw)["contents"]
+        if result.raw:
+            contents = json.loads(result.raw)
+            contents = contents['contents'] if 'contents' in contents else contents['content']
+        else:
+            contents = ''
+
         output = f'<div class="row"> <b>Rank</b>: {i+1} | <b>Document ID</b>: {result_id} | <b>Score</b>:{result_score:.2f}</div>'
 
         try:

pyserini/search/lucene/_impact_searcher.py

@@ -142,8 +142,9 @@ class LuceneImpactSearcher:
 
         encoded_query = self.encode(q)
 
-        jquery = JHashMap()
+        jquery = encoded_query
         if self.encoder_type == 'pytorch':
+            jquery = JHashMap()
             for (token, weight) in encoded_query.items():
                 if token in self.idf and self.idf[token] > self.min_idf:
                     jquery.put(token, JFloat(weight))