testbed / rebuild_faces.py
xspinners's picture
deepface
bc62936
from pymilvus import connections, utility
from dotenv import load_dotenv
import os
load_dotenv()
milvus_host = os.getenv("MILVUS_HOST", "milvus") # default localhost
milvus_port = os.getenv("MILVUS_PORT", "19530") # default 19530
connections.connect("default", host=milvus_host, port=int(milvus_port))
# Delete the collection
collection_name = "faces"
if utility.has_collection(collection_name):
utility.drop_collection(collection_name)
print(f"Collection '{collection_name}' deleted.")
else:
print(f"Collection '{collection_name}' does not exist.")