From a4191fa00c7ab7a678be9036fb86966eced80f55 Mon Sep 17 00:00:00 2001
From: Tinsae <tinsaekahsay@gmail.com>
Date: Tue, 5 May 2026 22:43:57 +0200
Subject: [PATCH] added loading and testing the model again

---
 architectures/Model.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/architectures/Model.py b/architectures/Model.py
index 50d506d..256b050 100644
--- a/architectures/Model.py
+++ b/architectures/Model.py
@@ -16,7 +16,6 @@ class Model(ABC):
 
     @abstractmethod
     def get(self):
-        # return the model
         pass
 
     def train(self, epochs, loader, rate):
@@ -66,7 +65,7 @@ class Model(ABC):
         save_dir = Path("trained_models")
         save_dir.mkdir(parents=True, exist_ok=True)
 
-        # 2. Determine filename (Default to class name if not provided)
+        # Determine filename (Default to class name if not provided)
         if filename is None:
             filename = f"{self.__class__.__name__.lower()}.pth"