Upload 7 files

app.py

@@ -1,7 +1,9 @@
 import os, random, gc, json, re
 from pathlib import Path
+import spaces  # must be imported before torch for ZeroGPU startup detection
 import gradio as gr
 import torch
+from train_style_lora import train_chunk
 from PIL import Image
 from rembg import remove, new_session
 from diffusers import (
@@ -361,6 +363,8 @@ with gr.Blocks(title="AI Character Studio ULTRA") as demo:
             seed = gr.Number(value=-1, precision=0, label="Seed (-1 = random)")
             generate = gr.Button("✨ GENERATE HIGH-QUALITY CHARACTER", variant="primary", elem_id="generate")
             clear = gr.Button("Clear GPU Cache")
+            train_lora = gr.Button("🔥 TRAIN LORA CHUNK (ZeroGPU)", variant="primary")
+            train_status = gr.Textbox(label="LoRA Training Status", interactive=False, lines=3)
             status = gr.Textbox(label="System Status", interactive=False)
 
         with gr.Column(scale=7):
@@ -374,6 +378,7 @@ with gr.Blocks(title="AI Character Studio ULTRA") as demo:
         [output, used_seed, prompt_view]
     )
     clear.click(clear_cache, outputs=status)
+    train_lora.click(train_chunk, outputs=train_status)
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch(css=CSS, theme=gr.themes.Base())

requirements.txt

@@ -6,7 +6,6 @@ peft>=0.13.0
 safetensors>=0.4.5
 huggingface_hub>=0.26.0
 bitsandbytes>=0.45.0
-xformers>=0.0.28; platform_system == "Linux"
 gradio>=5.0.0
 Pillow>=10.0.0
 rembg>=2.0.59