bug fix

app.py

@@ -92,7 +92,7 @@ def create_args(
 
 # Add this function to get checkpoint files
 def get_checkpoint_files():
-    unet_files_dir = Path("unetFiles")
+    unet_files_dir = Path("checkpoints/unetFiles")
     if not unet_files_dir.exists():
         return []
     return [f.name for f in unet_files_dir.glob("*.pt")]