Update app.py

app.py

@@ -9,6 +9,7 @@ import json
 from huggingface_hub import hf_hub_download, HfApi, create_repo, upload_file
 import tempfile
 import multiprocessing as mp
+import spaces  # ZeroGPU support
 
 # Training imports
 from peft import LoraConfig, get_peft_model
@@ -640,6 +641,7 @@ Trigger: `{trigger_word}`
         return False, str(e)
 
 
+@spaces.GPU(duration=600)  # 10 minutes max for training
 def train_lora(
     images, trigger_word, output_name, num_steps, learning_rate, lora_rank,
     resolution, batch_size, upload_to_hub_flag, hub_repo_name,
@@ -1023,7 +1025,7 @@ image = pipe("{trigger_word}, your prompt here", num_inference_steps=8, guidance
 # ============================================
 # Gradio UI with Comic Style
 # ============================================
-with gr.Blocks(css=COMIC_CSS, theme=gr.themes.Soft(), title="Z-IMAGE GEN/LORA") as demo:
+with gr.Blocks(title="Z-IMAGE GEN/LORA") as demo:
     
     # HOME Button
     gr.HTML("""
@@ -1173,4 +1175,4 @@ with gr.Blocks(css=COMIC_CSS, theme=gr.themes.Soft(), title="Z-IMAGE GEN/LORA")
 
 if __name__ == "__main__":
     mp.set_start_method('spawn', force=True)
-    demo.launch()
+    demo.launch(css=COMIC_CSS, theme=gr.themes.Soft(), ssr_mode=False)