Surajrgl commited on
Commit
1d8387c
·
1 Parent(s): d3f68b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -1,3 +1,12 @@
 
1
  import gradio as gr
2
 
3
- gr.load("models/sentence-transformers/multi-qa-mpnet-base-cos-v1").launch()
 
 
 
 
 
 
 
 
 
1
+ from sentence_transformers import SentenceTransformer
2
  import gradio as gr
3
 
4
+ model = SentenceTransformer('sentence-transformers/multi-qa-mpnet-base-cos-v1')
5
+
6
+
7
+ def generate_embedding(query):
8
+ return model.encode(query)
9
+
10
+ demo = gr.Interface(fn=generate_embedding, inputs="text", outputs="text")
11
+
12
+ demo.launch( show_api=True)