Update app.py
Browse files
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] #
|
|
|
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}"
|