swag

1 files changed, 3 insertions, 3 deletions

diff --git a/src/ocr.rs b/src/ocr.rs
index bc7e535..cde116f 100644
--- a/src/ocr.rs
+++ b/src/ocr.rs
@@ -22,12 +22,12 @@ fn x_matches(image: &GrayImage, template: &GrayImage) -> Vec<u32> {
     );
     match_values
         .enumerate_pixels()
-        .filter(|(_x, _y, pix)| pix.0[0] > 0.9)
+        .filter(|(_x, _y, pix)| pix.0[0] > 0.95)
         .map(|(x, _y, _pix)| x)
         .collect()
 }
 
-pub fn ocr(image: DynamicImage) -> u32 {
+pub fn ocr(image: DynamicImage) -> Option<u32> {
     let grayscale_image = image::imageops::grayscale(&image);
     let mut digit_x_positions: Vec<(u8, u32)> = (0..10)
         .flat_map(|i| {
@@ -46,5 +46,5 @@ pub fn ocr(image: DynamicImage) -> u32 {
         .map(|(i, _x)| format!("{}", i))
         .collect();
     dbg!(&digits);
-    digits.parse().unwrap()
+    digits.parse().ok()
 }