CASSE LA TETE

app.py

@@ -19,6 +19,8 @@ dino_v2_image_processor = AutoImageProcessor.from_pretrained("./dinov2-base")
 
 # MongoDB
 MONGO_URI = os.environ.get("MONGO_URI")
+mongo = MongoClient(MONGO_URI)
+db = mongo["xbgp"]
 
 
 def process_image(image):
@@ -29,10 +31,6 @@ def process_image(image):
     # This will include preprocessing the image, passing it through the model,
     # and then formatting the output (extracted features).
 
-    # MongoDB
-    mongo = MongoClient(MONGO_URI)
-    db = mongo["xbgp"]
-
     # Load the index
     with open("images.json", "r") as f:
         images = json.load(f)