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 = []