Gordon Weakliem commited on
Commit
0922458
·
1 Parent(s): dfe1df2

Add sentences and similarity function

Browse files
Files changed (1) hide show
  1. app.py +14 -9
app.py CHANGED
@@ -2,15 +2,20 @@ from sentence_transformers import SentenceTransformer, SimilarityFunction
2
 
3
  import streamlit as st
4
 
5
- st.
6
- x = st.slider('Select a value')
7
- st.write(x, 'squared is', x * x)
 
 
 
8
 
9
- model = SentenceTransformer(model_name, trust_remote_code=True)
10
- embeddings = model.encode(sentences, prompt_name="passage")
 
 
11
 
12
- similarity_fn_enum = getattr(SimilarityFunction, similarity_fn.upper())
13
- model.similarity_fn_name = similarity_fn_enum
14
 
15
- similarities = model.similarity(embeddings[0], embeddings[1])
16
- print(f"similarity: {similarities}")
 
2
 
3
  import streamlit as st
4
 
5
+ model_name = "nomic-ai/nomic-embed-text-v2-moe"
6
+ with st.form("embedding"):
7
+ sentence1 = st.text_inupt("Sentence 1:","Hello!")
8
+ sentence2 = st.text_inupt("Sentence 2:","¡Hola!")
9
+ sim_fun = st.selectbox('Similarity Function', ['COSINE', 'DOT_PRODUCT', 'EUCLIDEAN', 'MANHATTAN'])
10
+ calculate = st.form_submit_button('Calculate')
11
 
12
+ if calculate:
13
+ model = SentenceTransformer(model_name, trust_remote_code=True)
14
+ sentences = [sentence1, sentence2]
15
+ embeddings = model.encode(sentences, prompt_name="passage")
16
 
17
+ similarity_fn_enum = getattr(SimilarityFunction, sim_fun)
18
+ model.similarity_fn_name = similarity_fn_enum
19
 
20
+ similarities = model.similarity(embeddings[0], embeddings[1])
21
+ print(f"similarity: {similarities}")