Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,9 +27,9 @@ if not keys:
|
|
| 27 |
def get_random_key():
|
| 28 |
return random.choice(keys)
|
| 29 |
|
| 30 |
-
def generate_cache_key(prompt, width, height, seed, model_name
|
| 31 |
-
# Создаем уникальный ключ на основе всех
|
| 32 |
-
return hashlib.md5(f"{prompt}_{width}_{height}_{seed}_{model_name}
|
| 33 |
|
| 34 |
def scale_dimensions(width, height, max_width, max_height):
|
| 35 |
"""Масштабирует размеры изображения, сохраняя соотношение сторон, и округляет до чисел, кратных 8."""
|
|
@@ -118,7 +118,7 @@ def get_image(prompt):
|
|
| 118 |
api_key = get_random_key()
|
| 119 |
|
| 120 |
# Генерируем уникальный ключ для кэша
|
| 121 |
-
cache_key = generate_cache_key(prompt, width, height, seed, model_name
|
| 122 |
|
| 123 |
try:
|
| 124 |
image = generate_cached_image(cache_key, prompt, width, height, seed, model_name, api_key)
|
|
|
|
| 27 |
def get_random_key():
|
| 28 |
return random.choice(keys)
|
| 29 |
|
| 30 |
+
def generate_cache_key(prompt, width, height, seed, model_name):
|
| 31 |
+
# Создаем уникальный ключ на основе всех параметров, кроме ключа API
|
| 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."""
|
|
|
|
| 118 |
api_key = get_random_key()
|
| 119 |
|
| 120 |
# Генерируем уникальный ключ для кэша
|
| 121 |
+
cache_key = generate_cache_key(prompt, width, height, seed, model_name)
|
| 122 |
|
| 123 |
try:
|
| 124 |
image = generate_cached_image(cache_key, prompt, width, height, seed, model_name, api_key)
|