mehmetkeskinoglu commited on
Commit
ad692df
·
verified ·
1 Parent(s): 9f92ef3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -10
app.py CHANGED
@@ -1,19 +1,23 @@
1
  import gradio as gr
2
- from sentence_transformers import SentenceTransformer, util
 
 
3
 
 
4
  model = SentenceTransformer('sentence-transformers/paraphrase-mpnet-base-v2')
5
 
6
- def compute_similarity(text1, text2):
7
- embeddings = model.encode([text1, text2])
8
- similarity = util.cos_sim(embeddings[0], embeddings[1])
9
- return float(similarity)
10
 
 
11
  iface = gr.Interface(
12
- fn=compute_similarity,
13
- inputs=["text", "text"],
14
- outputs="number",
15
- title="Sentence Similarity with paraphrase-mpnet-base-v2",
16
- description="Compute the cosine similarity between two sentences."
17
  )
18
 
19
  iface.launch()
 
1
  import gradio as gr
2
+ from sentence_transformers import SentenceTransformer
3
+ import numpy as np
4
+ import json
5
 
6
+ # Modeli yükle
7
  model = SentenceTransformer('sentence-transformers/paraphrase-mpnet-base-v2')
8
 
9
+ # Embedding hesaplama fonksiyonu
10
+ def get_embedding(text):
11
+ embedding = model.encode(text)
12
+ return json.dumps(embedding.tolist()) # JSON formatında döndür
13
 
14
+ # Gradio arayüzü
15
  iface = gr.Interface(
16
+ fn=get_embedding,
17
+ inputs=gr.Textbox(lines=2, placeholder="Metni buraya girin..."),
18
+ outputs="text",
19
+ title="paraphrase-mpnet-base-v2 Embedding API",
20
+ description="Metni girin ve modelin embedding çıktısını JSON formatında alın."
21
  )
22
 
23
  iface.launch()