feat: :art: applying solutoion 2

utils/model.py

@@ -20,6 +20,11 @@ def create_effnetb2_model(num_class: int):
 
     """
 
+    # Change path for storing cache.
+    path_dir_cache = Path(".cache")
+    torch.hub.set_dir(path_dir_cache)
+    print(f"[DEBUG] Change Storing Cache in directory: {path_dir_cache}")
+
     # Get weights of ResNet50
     weights_effnetb2 = torchvision.models.EfficientNet_B2_Weights.IMAGENET1K_V1