Update app.py

app.py

@@ -247,15 +247,15 @@ Output: {result.stdout if result.stdout else 'Processing completed'}
                 str(train_script),
                 "-e", model_name,
                 "-sr", str(sample_rate),
-                "-f0", "1",  # Use F0
+                "-f0", "1",
                 "-bs", str(batch_size),
-                "-g", "0",  # GPU index (0 for CPU)
+                "-g", "0",
                 "-te", str(epochs),
-                "-se", "10",  # Save every N epochs
+                "-se", "10",
                 "-pg", str(self.rvc_dir / "pretrained" / "f0G40k.pth"),
                 "-pd", str(self.rvc_dir / "pretrained" / "f0D40k.pth"),
-                "-l", "0",  # No cache
-                "-c", "0"   # No cache
+                "-l", "0",
+                "-c", "0"
             ]
             
             progress(0.2, desc=f"Training {model_name}...")
@@ -292,26 +292,22 @@ Output: {result.stdout if result.stdout else 'Processing completed'}
                 if path.exists():
                     model_files.extend(list(path.glob("**/*.pth")))
             
-            # Also search for index files
             index_files = []
             for path in possible_paths:
                 if path.exists():
                     index_files.extend(list(path.glob("**/*.index")))
             
             if model_files or index_files:
-                # Create output directory
                 output_dir = self.workspace / model_name
                 output_dir.mkdir(exist_ok=True)
                 
-                # Copy model file
                 if model_files:
                     latest_model = max(model_files, key=lambda p: p.stat().st_mtime)
                     shutil.copy2(latest_model, output_dir / f"{model_name}.pth")
-                    model_size = latest_model.stat().st_size / (1024*1024)  # MB
+                    model_size = latest_model.stat().st_size / (1024*1024)
                 else:
                     model_size = 0
                 
-                # Copy index file
                 if index_files:
                     latest_index = max(index_files, key=lambda p: p.stat().st_mtime)
                     shutil.copy2(latest_index, output_dir / latest_index.name)
@@ -339,7 +335,6 @@ Output: {result.stdout if result.stdout else 'Processing completed'}
 🎉 Ready to download!
 """
             else:
-                # List what's in the log directory for debugging
                 debug_info = []
                 if log_dir.exists():
                     debug_info.append(f"Log directory exists: {log_dir}")