Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -117,21 +117,23 @@ def query(prompt, is_realistic):
|
|
117 |
image_bytes = response.content
|
118 |
image = Image.open(io.BytesIO(image_bytes))
|
119 |
|
|
|
|
|
120 |
if (is_realistic == True):
|
121 |
-
image = add_noise(image, intensity = randint(
|
122 |
-
image =
|
123 |
-
|
|
|
|
|
124 |
|
125 |
# Изменение насыщенности
|
126 |
enhancer = ImageEnhance.Color(image)
|
127 |
-
image = enhancer.enhance(randint(70,
|
128 |
|
129 |
# Изменение экспозиции (яркости)
|
130 |
enhancer = ImageEnhance.Brightness(image)
|
131 |
-
image = enhancer.enhance(randint(
|
132 |
-
|
133 |
-
enhancer = ImageEnhance.Contrast(image)
|
134 |
-
image = enhancer.enhance(randint(80,95) / 100)
|
135 |
|
136 |
print(f'\033[1mGeneration {key} completed!\033[0m ({prompt})')
|
137 |
return image
|
|
|
117 |
image_bytes = response.content
|
118 |
image = Image.open(io.BytesIO(image_bytes))
|
119 |
|
120 |
+
image = resize_and_crop(image, 480, 640)
|
121 |
+
|
122 |
if (is_realistic == True):
|
123 |
+
image = add_noise(image, intensity = randint(100, 150) / 10)
|
124 |
+
image = compress_image(image, randint(75, 85))
|
125 |
+
|
126 |
+
enhancer = ImageEnhance.Contrast(image)
|
127 |
+
image = enhancer.enhance(randint(75,85) / 100)
|
128 |
|
129 |
# Изменение насыщенности
|
130 |
enhancer = ImageEnhance.Color(image)
|
131 |
+
image = enhancer.enhance(randint(70,80) / 100)
|
132 |
|
133 |
# Изменение экспозиции (яркости)
|
134 |
enhancer = ImageEnhance.Brightness(image)
|
135 |
+
image = enhancer.enhance(randint(90,110) / 100)
|
136 |
+
|
|
|
|
|
137 |
|
138 |
print(f'\033[1mGeneration {key} completed!\033[0m ({prompt})')
|
139 |
return image
|