init

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import torch
 from sentence_transformers import SentenceTransformer, util
 
 def app():
@@ -42,10 +43,10 @@ def app():
 
         st.write(source_emb)
         st.write(sent_embs)
-        st.write(cos_sim)
+        st.write(cos_sim) #output tensor([[1.0000, 0.3624]])
         cosin_dict = {}
 
-        for i, cos in cos_sim[0]:
+        for i, cos in enumerate(torch.flatten(cos_sim)):
             cosin_dict[sentences[i]] = cos
         
         sorted_dict = dict(sorted(cosin_dict.items(), key=lambda item: item[1],reverse = True))