Spaces:
Running
Running
Update app.py
Browse files
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 |
# Генерируем уникальный ключ для кэша
|