liuwei commited on
Commit
51f25dc
·
1 Parent(s): 22c71f8
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -40,6 +40,8 @@ def app():
40
  sent_embs = model.encode(sentences, convert_to_tensor=True)
41
  cos_sim = util.cos_sim(source_emb, sent_embs)
42
 
 
 
43
  st.write(cos_sim)
44
  cosin_dict = {}
45
 
@@ -60,11 +62,13 @@ def app():
60
 
61
  with st.container():
62
  for sent, cos in embeddings:
63
- col1, col2 = st.columns(2)
64
  with col1:
65
- bar = st.progress(cos)
66
- with col2:
67
  st.text(sent)
 
 
 
 
68
 
69
  if button_clear:
70
  st.session_state.inputs.clear()
 
40
  sent_embs = model.encode(sentences, convert_to_tensor=True)
41
  cos_sim = util.cos_sim(source_emb, sent_embs)
42
 
43
+ st.write(source_emb)
44
+ st.write(sent_embs)
45
  st.write(cos_sim)
46
  cosin_dict = {}
47
 
 
62
 
63
  with st.container():
64
  for sent, cos in embeddings:
65
+ col1, col2, col3 = st.columns(3)
66
  with col1:
 
 
67
  st.text(sent)
68
+ with col2:
69
+ bar = st.progress(cos)
70
+ with col3:
71
+ st.text(cos)
72
 
73
  if button_clear:
74
  st.session_state.inputs.clear()