Update app.py

app.py

@@ -654,20 +654,7 @@ def train_model(data_path, epochs, batch_size, learning_rate, image_size, save_n
                     "text_encoder_state_dict": TEXT_ENCODER.state_dict(),
                     "config": CONFIG
                 }, ckpt_path)
-                logs.append(f"💾 Saved checkpoint at epoch {epoch+1}")
-                try:
-                    from huggingface_hub import HfApi
-                    api = HfApi()
-                    api.upload_file(
-                        path_or_fileobj=ckpt_path,
-                        path_in_repo=ckpt_path,
-                        repo_id="Spanicin/candlestick-diffusion",
-                        repo_type="space",
-                        token=os.environ.get("HF_TOKEN")
-                    )
-                    logs.append(f"☁️ Uploaded to repo")
-                except Exception as e:
-                    logs.append(f"⚠️ Upload failed: {e}")
+                print(f"💾 Saved checkpoint at epoch {epoch+1}")
         
         # Save model
         MODEL.eval()