22

app.py

@@ -46,28 +46,13 @@ def inference(
 
     scheduler = FlowMatchEulerDiscreteScheduler(num_train_timesteps=1000, shift=3)
     pipe.scheduler = scheduler
-
-    ### load input_image
-    
-
-    # input_image = input_image.convert("RGB").resize((1024, 1024))
-
-    # input_tensor = pil_image_to_torch_bgr(input_image).to(pipe.vae.device).to(pipe.vae.dtype)
-    input_tensor = pipe.image_processor.preprocess(input_image, height=height, width=width)
-    # input_tensor = image.to(dtype=torch.float32)
-    input_tensor = input_tensor.to(device=pipe.vae.device, dtype=pipe.vae.dtype)
-
-    # input_latents = _encode_vae_image(pipe=pipe, image=input_tensor, generator=generator)
-    # input_latents = pipe.vae.encode(input_tensor).latent_dist.sample()
-    # input_latents = input_latents * pipe.vae.config.scaling_factor
     
     image = pipe(
-        # latents=input_latents,
         prompt=prompt,
         guidance_scale=guidance_scale,
         num_inference_steps=num_inference_steps,
-        # width=width,
-        # height=height,
+        width=width,
+        height=height,
         generator=generator,
     ).images[0]
 
@@ -96,7 +81,6 @@ with gr.Blocks() as demo:
     with gr.Column():
         gr.HTML(read_file("static/header.html"))
     with gr.Column(elem_id="col-container"):
-        input_image = gr.Image(height=800,sources=['upload','clipboard'],image_mode='RGB', elem_id="image_upload", type="pil", label="Upload")
         with gr.Row():
             prompt = gr.Text(
                 label="Prompt",
@@ -146,7 +130,7 @@ with gr.Blocks() as demo:
                     minimum=0.0,
                     maximum=10.0,
                     step=0.1,
-                    value=0.0,
+                    value=5.0,
                 )
 
                 num_inference_steps = gr.Slider(
@@ -162,7 +146,6 @@ with gr.Blocks() as demo:
         triggers=[run_button.click, prompt.submit],
         fn=inference,
         inputs=[
-            input_image,
             prompt,
             seed,
             randomize_seed,

static/header.html

@@ -1,11 +1,11 @@
 <div style="text-align: center; max-width: 600px; margin: 0 auto;">
     <h1>
-        Z-Image [Turbo] [img2img]
+        Z Image Turbo: Fast & Versatile Image Generation
     </h1>
     <div class="grid-container" >
         <p>
-            Besides making new images, Z-Image is also good at changing existing ones. 
-            It can fix problems like weird fingers and faces, and it can also add cool artistic styles. 
+            Z-Image-Turbo is a distilled version that excels in photorealistic image generation, 
+            bilingual text rendering, and instruction adherence. 
             If you like my spaces, please support me by visiting <a href="https://aisudo.com/" target="_blank">AiSudo</a> for more image generation 😊
     </div>
 </div>