summerstay commited on
Commit
06aae6f
1 Parent(s): 3a3d683

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -0
app.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from sentence_transformers import SentenceTransformer
3
+
4
+ model_Q = SentenceTransformer('flax-sentence-embeddings/multi-QA_v1-mpnet-asymmetric-Q')
5
+
6
+ def getVectors(sentences):
7
+ vectors = []
8
+ splitSentences = sentences.split('/*/')
9
+ for sentence in sentences:
10
+ vectors.append(model_Q.encode(sentence))
11
+ return vectors
12
+
13
+ interface = gr.Interface(fn = getVectors,
14
+ inputs = "text",
15
+ outputs = ['text'],
16
+ title = 'get vectors',
17
+ description = 'get vectors for search')
18
+
19
+ interface.launch(inline = False)