Update app.py

app.py

@@ -72,10 +72,10 @@ print("YOLOWorld model loaded in {:.2f} seconds".format(time.time() - start_time
 
 print("Loading YOLO reference model...")
 start_time = time.time()
-reference_model_path = os.path.join(CACHE_DIR, "coin_det.pt")
+reference_model_path = os.path.join(CACHE_DIR, "best_coin.pt")
 if not os.path.exists(reference_model_path):
     print("Caching YOLO reference model to", reference_model_path)
-    shutil.copy("coin_det.pt", reference_model_path)
+    shutil.copy("best_coin.pt", reference_model_path)
 reference_detector_global = YOLO(reference_model_path)
 print("YOLO reference model loaded in {:.2f} seconds".format(time.time() - start_time))
 
@@ -123,7 +123,7 @@ def unload_and_reload_models():
     gc.collect()
     new_drawer_detector = YOLOWorld(os.path.join(CACHE_DIR, "yolov8x-worldv2.pt"))
     new_drawer_detector.set_classes(["box"])
-    new_reference_detector = YOLO(os.path.join(CACHE_DIR, "coin_det.pt"))
+    new_reference_detector = YOLO(os.path.join(CACHE_DIR, "best_coin.pt"))
     new_birefnet = AutoModelForImageSegmentation.from_pretrained(
         "zhengpeng7/BiRefNet", trust_remote_code=True, cache_dir=CACHE_DIR
     )
@@ -158,7 +158,7 @@ def yolo_detect(image: Union[str, Path, int, Image.Image, list, tuple, np.ndarra
 
 def detect_reference_square(img: np.ndarray):
     t = time.time()
-    res = reference_detector_global.predict(img, conf=0.15)
+    res = reference_detector_global.predict(img, conf=0.5)
     if not res or len(res) == 0 or len(res[0].boxes) == 0:
         raise ReferenceBoxNotDetectedError("Reference Coin not detected in the image.")
     print("Reference coin detection completed in {:.2f} seconds".format(time.time() - t))