Initial commit with folder contents

src/pipeline.py

@@ -95,9 +95,11 @@ def max_pixel_filter(image: Image) -> Image:
         # Convert the image to a numpy array
         img_array = np.array(image)
         # Find the maximum pixel value in the image
-        max_val = img_array.max()
+        # max_val = img_array.max()
+        max_val = img_array.min()
+
         # Reduce the maximum value to 1
-        img_array[img_array == max_val] -= 2
+        img_array[img_array == max_val] += 1
         # Convert the numpy array back to an image
         filtered_image = Image.fromarray(img_array)
         return filtered_image
@@ -956,8 +958,8 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
     # Warm-up runs
     for _ in range(4):
         pipeline(
-            prompt="a cat and a dog",
-            num_inference_steps=18
+            prompt="a girl and a woman",
+            num_inference_steps=14
         )
     pipeline.scheduler.prepare_loss()
     return pipeline
@@ -973,7 +975,7 @@ def infer(request: TextToImageRequest, pipeline: StableDiffusionXLPipeline) -> I
         width=request.width,
         height=request.height,
         generator=generator,
-        num_inference_steps=18,
+        num_inference_steps=14,
     ).images[0]
 
     filter_image = max_pixel_filter(image_0)