Chris4K commited on
Commit
7803d2d
·
verified ·
1 Parent(s): d15b2e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -393,7 +393,8 @@ def compare_embeddings(file, query, model_types, model_names, split_strategy, ch
393
  )
394
 
395
  # Storing embeddings into the results for future use
396
- result_embeddings = [doc['embedding'] for doc in results_raw] # Assuming each result has an embedding
 
397
 
398
  stats = calculate_statistics(results_raw, search_time, vector_store, num_tokens, embedding_model, query, top_k)
399
  stats["model"] = f"{model_type} - {model_name}"
 
393
  )
394
 
395
  # Storing embeddings into the results for future use
396
+ result_embeddings = [doc.metadata['embedding'] for doc in results_raw] # Adjust this based on the actual attribute names
397
+ # result_embeddings = [doc['embedding'] for doc in results_raw] # Assuming each result has an embedding
398
 
399
  stats = calculate_statistics(results_raw, search_time, vector_store, num_tokens, embedding_model, query, top_k)
400
  stats["model"] = f"{model_type} - {model_name}"