dengkane commited on
Commit
3e8419b
1 Parent(s): 42cce02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -0
app.py CHANGED
@@ -4,6 +4,27 @@ import streamlit as st
4
  import numpy as np
5
  import pandas as pd
6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  st.title('My first app')
8
 
9
  st.write("Here's our first attempt at using data to create a table:")
 
4
  import numpy as np
5
  import pandas as pd
6
 
7
+ from sentence_transformers import SentenceTransformer
8
+
9
+ model = SentenceTransformer('moka-ai/m3e-base')
10
+
11
+ #Our sentences we like to encode
12
+ sentences = [
13
+ '* Moka 此文本嵌入模型由 MokaAI 训练并开源,训练脚本使用 uniem',
14
+ '* Massive 此文本嵌入模型通过**千万级**的中文句对数据集进行训练',
15
+ '* Mixed 此文本嵌入模型支持中英双语的同质文本相似度计算,异质文本检索等功能,未来还会支持代码检索,ALL in one'
16
+ ]
17
+
18
+ #Sentences are encoded by calling model.encode()
19
+ embeddings = model.encode(sentences)
20
+
21
+ #Print the embeddings
22
+ for sentence, embedding in zip(sentences, embeddings):
23
+ st.write("Sentence:", sentence)
24
+ st.write("Embedding:", embedding)
25
+ st.write("")
26
+
27
+
28
  st.title('My first app')
29
 
30
  st.write("Here's our first attempt at using data to create a table:")