Kazel
commited on
Commit
·
97177b4
1
Parent(s):
26b5c93
change
Browse files- app.py +1 -1
- milvus_manager.py +1 -1
app.py
CHANGED
@@ -80,7 +80,7 @@ class PDFSearchApp:
|
|
80 |
|
81 |
#clear files for next consec upload after loop is complete
|
82 |
files = []
|
83 |
-
return f"Uploaded and extracted
|
84 |
except Exception as e:
|
85 |
return f"Error processing PDF: {str(e)}"
|
86 |
|
|
|
80 |
|
81 |
#clear files for next consec upload after loop is complete
|
82 |
files = []
|
83 |
+
return f"Uploaded and extracted all pages"
|
84 |
except Exception as e:
|
85 |
return f"Error processing PDF: {str(e)}"
|
86 |
|
milvus_manager.py
CHANGED
@@ -65,7 +65,7 @@ class MilvusManager:
|
|
65 |
collections = self.client.list_collections()
|
66 |
|
67 |
# Set search parameters (here, using Inner Product metric).
|
68 |
-
search_params = {"metric_type":
|
69 |
|
70 |
# Set to store unique (doc_id, collection_name) pairs across all collections.
|
71 |
doc_collection_pairs = set()
|
|
|
65 |
collections = self.client.list_collections()
|
66 |
|
67 |
# Set search parameters (here, using Inner Product metric).
|
68 |
+
search_params = {"metric_type": "COSINE", "params": {}} #default metric type is "IP"
|
69 |
|
70 |
# Set to store unique (doc_id, collection_name) pairs across all collections.
|
71 |
doc_collection_pairs = set()
|