issue fix

app.py

@@ -129,7 +129,6 @@ def generate_community_image(prompt, model_name, seed, num_steps, guidance_scale
             base_model_id,
             torch_dtype=dtype,
             use_safetensors=True,
-            variant="fp16" if device == "cuda" else None
         ).to(device)
         
         pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
@@ -297,17 +296,15 @@ def inpaint_image(prompt, init_image, mask_image, seed, num_steps, guidance_scal
             generator=generator,
         ).images[0]
         
-        # Create a grid of the original image, mask, and result
-        grid = make_image_grid([init_image, mask_image, image], rows=1, cols=3)
-        
         # Cleanup
         del pipe
         cleanup_memory()
         
-        return grid, "Image generated successfully!"
+        # Return individual images instead of a grid
+        return init_image, mask_image, image, "Image generated successfully!"
     except Exception as e:
         cleanup_memory()
-        return None, f"Error generating image: {str(e)}"
+        return None, None, None, f"Error generating image: {str(e)}"
 
 def generate_animation(prompt, seed, num_steps, guidance_scale, eta, num_frames, motion_scale):
     try:
@@ -418,9 +415,13 @@ with gr.Blocks(title="TCD-SDXL Image Generator") as demo:
                     inpaint_eta = gr.Slider(minimum=0, maximum=1, value=0.3, label="Eta")
                     inpaint_strength = gr.Slider(minimum=0, maximum=1, value=0.99, label="Strength")
                     inpaint_button = gr.Button("Inpaint")
-                with gr.Column():
-                    inpaint_output = gr.Image(label="Result (Original | Mask | Generated)")
                     inpaint_status = gr.Textbox(label="Status", interactive=False)
+                with gr.Column():
+                    # Display individual images in a row
+                    with gr.Row():
+                        inpaint_output_original = gr.Image(label="Original")
+                        inpaint_output_mask = gr.Image(label="Mask")
+                        inpaint_output_generated = gr.Image(label="Generated")
             
             inpaint_button.click(
                 fn=inpaint_image,
@@ -428,7 +429,8 @@ with gr.Blocks(title="TCD-SDXL Image Generator") as demo:
                     inpaint_prompt, init_image, mask_image, inpaint_seed,
                     inpaint_steps, inpaint_guidance, inpaint_eta, inpaint_strength
                 ],
-                outputs=[inpaint_output, inpaint_status]
+                # Map function outputs to individual image components and status
+                outputs=[inpaint_output_original, inpaint_output_mask, inpaint_output_generated, inpaint_status]
             )
             
         with gr.TabItem("Community Models"):