Upload app.py

app.py

@@ -1295,7 +1295,14 @@ def train_model_lora(
 
         # ── Dataset: carregar do disco ou tokenizar ────────────────────────
         tokenized_path = os.path.join(output_dir, "tokenized_dataset")
-
+#----------------------------------------------------------------------------
+        # Depois de tokenizar o dataset
+        if os.path.exists(tokenized_path):
+            HF_BUCKET_DATASET = "hf://buckets/Fedir-Ilina/Train_CPU/tokenized_dataset"  # definir nome do bucket
+            upload_to_bucket(tokenized_path, HF_BUCKET_DATASET)
+            log_success("Dataset tokenizado enviado ao Bucket Hugging Face!")
+            training_logs.append("Dataset tokenizado enviado ao Bucket Hugging Face.")
+#----------------------------------------------------------------------------
         if os.path.exists(tokenized_path):
             log_info("Dataset tokenizado encontrado no disco. A carregar...")
             tokenized_datasets = load_from_disk(tokenized_path)