codeblacks commited on
Commit
e4a6428
·
verified ·
1 Parent(s): dbf6cc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,11 +1,13 @@
1
  from sentence_transformers import SentenceTransformer
2
  import gradio as gr
 
3
 
4
  # Load the pre-trained model
5
  model = SentenceTransformer('all-MiniLM-L6-v2')
6
 
7
  def get_embeddings(sentences):
8
- embeddings = model.encode(sentences, convert_to_tensor=True)
 
9
  return embeddings.tolist()
10
 
11
  # Define the Gradio interface
 
1
  from sentence_transformers import SentenceTransformer
2
  import gradio as gr
3
+ import ast
4
 
5
  # Load the pre-trained model
6
  model = SentenceTransformer('all-MiniLM-L6-v2')
7
 
8
  def get_embeddings(sentences):
9
+ chunks = ast.literal_eval(sentences)
10
+ embeddings = model.encode(chunks, convert_to_tensor=True)
11
  return embeddings.tolist()
12
 
13
  # Define the Gradio interface