22

app.py

@@ -23,7 +23,6 @@ spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant
 
 @spaces.GPU
 def inference(
-    input_image,
     prompt,
     seed=42,
     randomize_seed=True,
@@ -33,9 +32,6 @@ def inference(
     num_inference_steps=8,
     progress=gr.Progress(track_tqdm=True),
 ):
-    if input_image is None:
-        print("Error: input_image is empty.")
-        return None
     
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)