EmoCube commited on
Commit
100c04b
·
verified ·
1 Parent(s): 24910cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -8
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(75, 125) / 10)
122
- image = resize_and_crop(image, 480, 640)
123
- image = compress_image(image, randint(75, 95))
 
 
124
 
125
  # Изменение насыщенности
126
  enhancer = ImageEnhance.Color(image)
127
- image = enhancer.enhance(randint(70,90) / 100)
128
 
129
  # Изменение экспозиции (яркости)
130
  enhancer = ImageEnhance.Brightness(image)
131
- image = enhancer.enhance(randint(100,110) / 100)
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