feat: autoencoder + updated trainers + cleaned up process to allow using autoencoder

main.py

@@ -1,5 +1,5 @@
 from src.args import parse_arguments
-from src.process import compress
+from src.process import compress, decompress
 from src.train import train
 from src.utils import determine_device
 
@@ -38,12 +38,21 @@ def main():
 
         case 'compress':
             compress(device=device,
+                     model_name=args.model,
                      model_path=args.model_load_path,
                      input_file=args.input_file,
                      output_file=args.output_file,
                      context_length=args.context
                      )
-
+        case 'decompress':
+            decompress(
+                device=device,
+                model_name=args.model,
+                model_path=args.model_load_path,
+                input_file=args.input_file,
+                output_file=args.output_file,
+                context_length=args.context
+            )
         case _:
             raise NotImplementedError(f"Mode {args.mode} is not implemented yet")