init

app.py

@@ -64,7 +64,8 @@ def app():
         with st.container():
             for sent, cos in embeddings.items():
                 col1, col2, col3 = st.columns(3)
-                cos_value = float(cos.split('(')[1].split(')')[0])
+                st.write(type(cos))
+                cos_value = cos.item()
                 with col1:
                     st.text(sent)
                 with col2: