fix output

app.py

@@ -169,20 +169,15 @@ def generate_image(prompt, negative_prompt, width, height, steps, cfg, seed):
             vae=get_value_at_index(vaeloader_80, 0),
         )
 
-        # Convert tensor to PIL Image
-        image_tensor = get_value_at_index(vaedecode_79, 0)
-        # Convert from [0,1] to [0,255] range and ensure it's in the right format
-        image_np = (image_tensor.cpu().numpy() * 255).astype(np.uint8)
-        
-        # Ensure we have the right shape (height, width, channels)
-        if len(image_np.shape) == 4:  # (batch, channel, height, width)
-            image_np = image_np[0]  # Remove batch dimension
-        if len(image_np.shape) == 3:  # (channel, height, width)
-            image_np = np.transpose(image_np, (1, 2, 0))  # Convert to (height, width, channel)
+        # Save image using SaveImage node with simple string prefix
+        saved = saveimage.save_images(
+            filename_prefix="Chroma_Generated",
+            images=get_value_at_index(vaedecode_79, 0),
+        )
         
-        # Convert to PIL Image
-        image = Image.fromarray(image_np)
-        return image
+        # Return the path to the saved image
+        saved_path = f"output/{saved['ui']['images'][0]['filename']}"
+        return saved_path
 
 # Create Gradio interface
 with gr.Blocks() as app: