Update app.py

app.py

@@ -222,40 +222,6 @@ def log_message(output_log, msg):
 # 🧠 Train model to expand short prompts into long ones
 # =====================================================
 @spaces.GPU(duration=300)
-import torch
-from datasets import load_dataset
-from transformers import (
-    AutoTokenizer,
-    AutoModelForCausalLM,
-    Trainer,
-    TrainingArguments,
-)
-from peft import LoraConfig, get_peft_model, TaskType
-import os
-
-# =====================================================
-# 🔧 Utility logging
-# =====================================================
-def log_message(output_log, msg):
-    print(msg)
-    output_log.append(msg)
-
-def start_async_upload(output_dir, hf_repo, output_log):
-    from huggingface_hub import upload_folder
-    try:
-        upload_folder(
-            repo_id=hf_repo,
-            folder_path=output_dir,
-            repo_type="model",
-            commit_message="Upload fine-tuned model"
-        )
-        log_message(output_log, f"☁️ Model uploaded to {hf_repo}")
-    except Exception as e:
-        log_message(output_log, f"⚠️ Upload failed: {e}")
-
-# =====================================================
-# 🧠 GPU-safe training for short→long prompt expansion
-# =====================================================
 def train_model(
     base_model: str,
     dataset_name: str,