22

app.py

@@ -20,6 +20,7 @@ MAX_IMAGE_SIZE = 1280
 MODEL_LOCAL = "models/Z-Image-Turbo/"
 # curl -L -o Z-Image-Turbo-Fun-Controlnet-Union.safetensors https://huggingface.co/alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union/resolve/main/Z-Image-Turbo-Fun-Controlnet-Union.safetensors
 TRANSFORMER_LOCAL = "models/Z-Image-Turbo-Fun-Controlnet-Union.safetensors"
+INFERENCE_CNT = 0
 
 weight_dtype = torch.bfloat16
 
@@ -97,6 +98,10 @@ def inference(
     num_inference_steps=8,
     progress=gr.Progress(track_tqdm=True),
 ):
+    INFERENCE_CNT = INFERENCE_CNT + 1
+    print("INFERENCE_CNT: " + INFERENCE_CNT)
+
+
     # process image
     print("DEBUG: process image")
     if input_image is None: