Update app.py

app.py

@@ -647,14 +647,7 @@ def train_model(data_path, epochs, batch_size, learning_rate, image_size, save_n
             avg_loss = epoch_loss / len(train_loader)
             logs.append(f"Epoch {epoch+1}/{epochs}: loss = {avg_loss:.4f}")
 
-            if (epoch + 1) % 10 == 0:
-                ckpt_path = f"checkpoints/{save_name}_epoch{epoch+1}.pt"
-                torch.save({
-                    "model_state_dict": MODEL.state_dict(),
-                    "text_encoder_state_dict": TEXT_ENCODER.state_dict(),
-                    "config": CONFIG
-                }, ckpt_path)
-                print(f"💾 Saved checkpoint at epoch {epoch+1}")
+        
         
         # Save model
         MODEL.eval()