Gowthamr23 commited on
Commit
63f174e
·
verified ·
1 Parent(s): bab4aa8

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. .env.example +1 -1
  2. app/config.py +1 -1
.env.example CHANGED
@@ -12,7 +12,7 @@ SUPABASE_ANON_KEY=your_supabase_anon_key_here
12
 
13
  # Object Detection (YOLOv8n ONNX) — crops detected objects before CLIP
14
  ENABLE_OBJECT_DETECTION=false
15
- YOLO_MODEL_ID=ultralytics/yolov8n
16
  YOLO_MODEL_FILENAME=yolov8n.onnx
17
  DETECTION_CONFIDENCE_THRESHOLD=0.25
18
  DETECTION_IOU_THRESHOLD=0.45
 
12
 
13
  # Object Detection (YOLOv8n ONNX) — crops detected objects before CLIP
14
  ENABLE_OBJECT_DETECTION=false
15
+ YOLO_MODEL_ID=Kalray/yolov8
16
  YOLO_MODEL_FILENAME=yolov8n.onnx
17
  DETECTION_CONFIDENCE_THRESHOLD=0.25
18
  DETECTION_IOU_THRESHOLD=0.45
app/config.py CHANGED
@@ -17,7 +17,7 @@ SIMILARITY_THRESHOLD = 0.65
17
 
18
  # Object Detection (YOLOv8n ONNX)
19
  ENABLE_OBJECT_DETECTION = os.getenv("ENABLE_OBJECT_DETECTION", "false").lower() == "true"
20
- YOLO_MODEL_ID = os.getenv("YOLO_MODEL_ID", "ultralytics/yolov8n")
21
  YOLO_MODEL_FILENAME = os.getenv("YOLO_MODEL_FILENAME", "yolov8n.onnx")
22
  DETECTION_CONFIDENCE_THRESHOLD = float(os.getenv("DETECTION_CONFIDENCE_THRESHOLD", "0.25"))
23
  DETECTION_IOU_THRESHOLD = float(os.getenv("DETECTION_IOU_THRESHOLD", "0.45"))
 
17
 
18
  # Object Detection (YOLOv8n ONNX)
19
  ENABLE_OBJECT_DETECTION = os.getenv("ENABLE_OBJECT_DETECTION", "false").lower() == "true"
20
+ YOLO_MODEL_ID = os.getenv("YOLO_MODEL_ID", "Kalray/yolov8")
21
  YOLO_MODEL_FILENAME = os.getenv("YOLO_MODEL_FILENAME", "yolov8n.onnx")
22
  DETECTION_CONFIDENCE_THRESHOLD = float(os.getenv("DETECTION_CONFIDENCE_THRESHOLD", "0.25"))
23
  DETECTION_IOU_THRESHOLD = float(os.getenv("DETECTION_IOU_THRESHOLD", "0.45"))