22

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import numpy as np
-import random, json, spaces, torch
+import random, json, spaces, torch, time
 from diffusers import AutoencoderKL, FlowMatchEulerDiscreteScheduler
 from videox_fun.pipeline import ZImageControlPipeline
 from videox_fun.models import ZImageControlTransformer2DModel
@@ -20,7 +20,6 @@ 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
 
@@ -98,9 +97,8 @@ def inference(
     num_inference_steps=8,
     progress=gr.Progress(track_tqdm=True),
 ):
-    INFERENCE_CNT = INFERENCE_CNT + 1
-    print("INFERENCE_CNT: " + INFERENCE_CNT)
-
+    
+    print("timestamp: " + time.time())
 
     # process image
     print("DEBUG: process image")