Add another model and make it a configurable parameter (#3)

* Add files via upload * allow model to be configurable with environment variable * Update README.md * Update README.md
laurynas · Feb 8, 2025 · 8bf263a · 8bf263a
1 parent 1283f1f
commit 8bf263a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -65,6 +65,8 @@ Converting to onnx format
 yolo export model=best.pt format=onnx
 ```
 
+There is a secondary model added from another [roboflow dataset](https://universe.roboflow.com/numbers-for-electric-meters/white-numbers) which some users have found more successful. That can be activated by adding the environment variable `MODEL_PATH=models/white-numbers.v2i.yolov8.onnx`
+
 ## ESPHome
 
 ESP32-Cam device [configuration examples](https://github.com/laurynas/esphome-devices/).

diff --git a/app.py b/app.py
@@ -1,4 +1,5 @@
 import json
+import os
 from PIL import Image
 from src.digitizer import Digitizer
 from src.routing import IdentifierConverter
@@ -8,7 +9,7 @@
 from glob import glob
 from flask import Flask, request, send_file
 
-model = 'models/yolov8-detect-20240229.onnx'
+model = os.getenv("MODEL_PATH", "models/yolov8-detect-20240229.onnx")
 data_dir = 'data/'
 
 digitizer = Digitizer(model)

diff --git a/models/white-numbers.v2i.yolov8.onnx b/models/white-numbers.v2i.yolov8.onnx