Update app.py

app.py

@@ -3,7 +3,17 @@ import spaces
 import gradio as gr
 from diffusers import DiffusionPipeline
 from optimization import optimize_pipeline_ 
-
+import spaces
+from dataclasses import dataclass
+import json
+import logging
+import os
+import random
+import re
+import sys
+import warnings
+ENABLE_COMPILE = os.environ.get("ENABLE_COMPILE", "true").lower() == "true"
+ENABLE_WARMUP = os.environ.get("ENABLE_WARMUP", "true").lower() == "true"
 
 # Load the pipeline once at startup
 print("Loading Z-Image-Turbo pipeline...")