Spaces:

dnth
/

blood-cell-counter

Sleeping

dnth commited on Mar 6, 2025

Commit

50b2c95

verified ·

1 Parent(s): e554661

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ from PIL import Image, ImageDraw
 # Use absolute paths instead of relative paths
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-MODEL_PATH = os.path.join(BASE_DIR, "models/deim-blood-cell-detection_nano.onnx")
 CLASS_NAMES_PATH = os.path.join(BASE_DIR, "models/classes.txt")
 def resize_with_aspect_ratio(image, size, interpolation=Image.BILINEAR):

 # Use absolute paths instead of relative paths
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+MODEL_PATH = os.path.join(BASE_DIR, "models/deim-blood-cell-detection_small.onnx")
 CLASS_NAMES_PATH = os.path.join(BASE_DIR, "models/classes.txt")
 def resize_with_aspect_ratio(image, size, interpolation=Image.BILINEAR):