Spaces:
Sleeping
Sleeping
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 |
# Генерируем уникальный ключ для кэша
|