Test

app.py

@@ -65,6 +65,8 @@ def chat():
 def get_chat_response(text):
     query_vector = model.encode(text_preprocessing(text)).astype(float)
     results = util.pytorch_cos_sim(query_vector, docs)
+    print(results.shape)
     top_n = 3
     sort_idx = results.argsort(descending=True, axis=1)[0][:top_n]
+    print(sort_idx)
     return df.iloc[sort_idx]