Automatically create the dir

app.py

@@ -19,9 +19,18 @@ use_cpu = not torch.cuda.is_available()
 device = "cpu" if use_cpu else "cuda"
 
 # Custom file cache decorator
+import os
+import pickle
+
 def file_cache(file_path):
     def decorator(func):
         def wrapper(*args, **kwargs):
+            # Ensure the directory exists
+            dir_path = os.path.dirname(file_path)
+            if not os.path.exists(dir_path):
+                os.makedirs(dir_path, exist_ok=True)
+                print(f"Created directory {dir_path}")
+
             # Check if the file already exists
             if os.path.exists(file_path):
                 # Load from cache