init

app.py

@@ -34,12 +34,13 @@ def app():
     button_generate = st.button("计算")
     button_clear = st.button("清空")
 
-    def transformer(sentences):
+    def transformer(source_text, sentences):
         model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', device="cpu")
         source_emb = model.encode(source_text, convert_to_tensor=True)
         sent_embs = model.encode(sentences, convert_to_tensor=True)
         cos_sim = util.cos_sim(source_emb, sent_embs)
 
+        st.write(cos_sim)
         cosin_dict = {}
 
         for i, cos in cos_sim[0]:
@@ -51,12 +52,16 @@ def app():
 
     if button_generate:
         # embeddings
-        embeddings = transformer(st.session_state.inputs)
+        embeddings = transformer([source_text], st.session_state.inputs)
 
         # 显示生成的文本
         st.write(embeddings)
         #output_text.success(generated_text)
 
+        with st.container():
+            for sent, cos in embeddings:
+                st.
+
     if button_clear:
         st.session_state.inputs.clear()
         del st.session_state["inputs"]
@@ -64,6 +69,8 @@ def app():
         source_text = ''
         st.experimental_rerun()
 
+   
+
 if __name__ == "__main__":
     # 运行应用程序
     app()