feat: Updated API/database.py

Devasy23 · Mar 12, 2024 · bcbca6b · bcbca6b
1 parent c692e31
commit bcbca6b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/API/database.py b/API/database.py
@@ -4,7 +4,8 @@
 
 
 class Database:
-    def __init__(self, uri="mongodb://localhost:27017/", db_name="ImageDB"):
+    def __init__(self, db_name="ImageDB", use_local=False):
+        uri = "mongodb://localhost:27017/" if use_local else "your_cloud_mongodb_atlas_uri"
         self.client = MongoClient(uri)
         self.db = self.client[db_name]