codeblacks commited on
Commit
9bb4ce0
·
verified ·
1 Parent(s): ef16b6d

Update app.py

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