Rooni commited on
Commit
4e84a67
·
verified ·
1 Parent(s): d8b1d45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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, api_key):
31
- # Создаем уникальный ключ на основе всех параметров
32
- return hashlib.md5(f"{prompt}_{width}_{height}_{seed}_{model_name}_{api_key}".encode()).hexdigest()
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, api_key)
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)