Rooni commited on
Commit
0402965
·
verified ·
1 Parent(s): 1e5c4a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -32,7 +32,7 @@ def generate_cache_key(prompt, width, height, seed, model_name):
32
  return hashlib.md5(f"{prompt}_{width}_{height}_{seed}_{model_name}".encode()).hexdigest()
33
 
34
  def scale_dimensions(width, height, max_width, max_height):
35
- """Масштабирует размеры изображения, сохраняя соотношение сторон."""
36
  aspect_ratio = width / height
37
  if width > max_width or height > max_height:
38
  if width / max_width > height / max_height:
@@ -41,6 +41,10 @@ def scale_dimensions(width, height, max_width, max_height):
41
  else:
42
  height = max_height
43
  width = int(height * aspect_ratio)
 
 
 
 
44
  return width, height
45
 
46
  @cached(cache)
@@ -98,7 +102,7 @@ def get_image(prompt):
98
  app.logger.error(f"Error translating prompt: {e}")
99
  return send_error_image()
100
 
101
- # Масштабируем размеры изображения, если они превышают максимальные значения
102
  width, height = scale_dimensions(width, height, MAX_WIDTH, MAX_HEIGHT)
103
 
104
  # Генерируем уникальный ключ для кэша
 
32
  return hashlib.md5(f"{prompt}_{width}_{height}_{seed}_{model_name}".encode()).hexdigest()
33
 
34
  def scale_dimensions(width, height, max_width, max_height):
35
+ """Масштабирует размеры изображения, сохраняя соотношение сторон, и округляет до чисел, кратных 8."""
36
  aspect_ratio = width / height
37
  if width > max_width or height > max_height:
38
  if width / max_width > height / max_height:
 
41
  else:
42
  height = max_height
43
  width = int(height * aspect_ratio)
44
+
45
+ # Округляем до ближайших чисел, кратных 8
46
+ width = (width + 3) // 8 * 8
47
+ height = (height + 3) // 8 * 8
48
  return width, height
49
 
50
  @cached(cache)
 
102
  app.logger.error(f"Error translating prompt: {e}")
103
  return send_error_image()
104
 
105
+ # Масштабируем размеры изображения, если они превышают максимальные значения, и округляем до чисел, кратных 8
106
  width, height = scale_dimensions(width, height, MAX_WIDTH, MAX_HEIGHT)
107
 
108
  # Генерируем уникальный ключ для кэша