add gaussian blur on rpogress images

handler.py

@@ -10,6 +10,8 @@ from pathlib import Path
 from typing import Union
 from concurrent.futures import ThreadPoolExecutor
 import numpy as np
+from PIL import ImageFilter
+
 
 import torch
 from diffusers import (
@@ -250,6 +252,10 @@ class EndpointHandler:
                 img_data = self.pipe.decode_latents(latents)
                 img_data = (img_data.squeeze() * 255).astype(np.uint8)
                 img = Image.fromarray(img_data, "RGB")
+
+                # Apply a blur to the image
+                img = img.filter(ImageFilter.GaussianBlur(radius=5))
+
                 # print(img_data)
             else:
                 # pil object
@@ -370,7 +376,7 @@ class EndpointHandler:
                     callback=lambda step, timestep, latents: self.progress_callback(
                         step, timestep, latents, request_id, "progress"
                     ),
-                    callback_steps=8,  #  The frequency at which the callback function is called.
+                    callback_steps=5,  #  The frequency at which the callback function is called.
                     # output_type="pt",
                 ).images[0]