From 9846ea8c1078d730b8d6154faef20532031fcd7a Mon Sep 17 00:00:00 2001
From: Nabil Ould Hamou <ouldhamounabil@gmail.com>
Date: Fri, 31 Jan 2025 23:23:10 +0100
Subject: [PATCH] Fixed a typo

---
 predict.py | 2 +-
 train.py   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/predict.py b/predict.py
index d9cb773..fb5c38b 100644
--- a/predict.py
+++ b/predict.py
@@ -10,7 +10,7 @@ if torch.cuda.is_available():
 elif torch.mps.is_available():
     DEVICE = torch.device("mps")
 else:
-    DEIVCE = torch.device("cpu")
+    DEVICE = torch.device("cpu")
 
 def predict_image(image_path, model):
     img = cv2.imread(image_path)
diff --git a/train.py b/train.py
index 5f1c3ac..f61d3c8 100644
--- a/train.py
+++ b/train.py
@@ -15,7 +15,7 @@ if torch.cuda.is_available():
 elif torch.mps.is_available():
     DEVICE = torch.device("mps")
 else:
-    DEIVCE = torch.device("cpu")
+    DEVICE = torch.device("cpu")
 
 def load_images_from_folder(folder, label):
     data = []