added loading and testing the model again

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"