codeblacks commited on
Commit
19343df
·
verified ·
1 Parent(s): 0b76df3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
8
  def generate_embeddings(chunks):
9
  embeddings = embedding_model.encode(chunks, convert_to_tensor=False)
10
  shape= embeddings.shape
11
- return embeddings, shape # Convert tensor to list for Gradio
12
 
13
  # Define the Gradio interface
14
  interface = gr.Interface(
 
8
  def generate_embeddings(chunks):
9
  embeddings = embedding_model.encode(chunks, convert_to_tensor=False)
10
  shape= embeddings.shape
11
+ return embeddings, shape # Convert tensor to list for Gradio
12
 
13
  # Define the Gradio interface
14
  interface = gr.Interface(