Update app.py

app.py

@@ -15,6 +15,8 @@ def last_token_pool(last_hidden_states: Tensor,
         return last_hidden_states[torch.arange(batch_size, device=last_hidden_states.device), sequence_lengths]
 
 def get_similarity_scores(queries:list, passages:list, model, tokenizer):
+    print("queries", queries)
+    print("passages", passages)
     tokenizer.add_eos_token = True
 
     max_length = 4096
@@ -28,9 +30,11 @@ def get_similarity_scores(queries:list, passages:list, model, tokenizer):
     return scores.tolist()
 
 def similarity_ui(keyNames, fields):
+    print("keynames", keyNames)
+    print("fields", fields)
     task = 'Given a keyName, find similarity score against provided fields'
-    queries = keyNames
-    passages = fields
+    queries = [keyNames]
+    passages = [fields]
 
     scores = get_similarity_scores(queries, passages, model, tokenizer)
     return {'Similarity Scores': scores}