add contrast

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import QuckDrawGAN as qd
 import numpy as np
-from PIL import Image
+from PIL import Image, ImageFilter, ImageEnhance
 
 # Загрузим модель
 generator_file = r'pretrained_output/models/generator.pt'
@@ -28,6 +28,13 @@ def generate_image(n_images=16, seed=""):
     pil_image = Image.fromarray(normalized_image.astype(np.uint8))  # Преобразуем в uint8 для отображения
     pil_image = pil_image.resize((256, 256), Image.Resampling.LANCZOS)  # Ресайз изображения до 256x256
 
+    # Применяем размытие
+    pil_image = pil_image.filter(ImageFilter.GaussianBlur(radius=1))  # Фильтр размытия с радиусом 2
+
+    # Применяем контрастность
+    enhancer = ImageEnhance.Contrast(pil_image)  # Создаем объект для изменения контрастности
+    pil_image = enhancer.enhance(3)  # Увеличиваем контрастность на 1.5
+
     return pil_image
 
 # Создаем интерфейс Gradio с помощью Blocks (для большей гибкости)