Update app.py

app.py

@@ -105,7 +105,7 @@ def process_image(
             tile_diffusion_stride=None,
             start_steps=999,
             start_point='lr',
-            use_vae_encode_condition=False,
+            use_vae_encode_condition=True,  # Changed to True
             sample_times=1
         )
 
@@ -138,6 +138,10 @@ def process_image(
         # Generate image
         with torch.no_grad():
             try:
+                # First encode the image with VAE
+                image_tensor = pipeline.image_processor.preprocess(validation_image)
+                image_tensor = image_tensor.unsqueeze(0).to(device="cuda", dtype=torch.float32)
+                
                 inference_time, output = pipeline(
                     args.t_max,
                     args.t_min,
@@ -155,7 +159,7 @@ def process_image(
                     conditioning_scale=args.conditioning_scale,
                     start_steps=args.start_steps,
                     start_point=args.start_point,
-                    use_vae_encode_condition=args.use_vae_encode_condition,
+                    use_vae_encode_condition=True,  # Set to True
                 )
             except Exception as e:
                 print(f"Pipeline execution error: {str(e)}")
@@ -180,6 +184,7 @@ def process_image(
         traceback.print_exc()
         return None
 
+
 # Create Gradio interface
 demo = gr.Interface(
     fn=process_image,