Kvikontent commited on
Commit
d671906
·
1 Parent(s): bace055

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -46,8 +46,7 @@ gan.compile(loss='binary_crossentropy', optimizer=Adam(learning_rate=0.0002, bet
46
  # Функция для генерации изображений на основе запроса
47
  def generate_images(prompt):
48
  num_images = 10
49
- prompt_vector = preprocess_prompt(prompt) # Предобработка запроса, если необходимо
50
- noise = np.repeat([prompt_vector], num_images, axis=0) # Генерация шума на основе запроса
51
  generated_images = generator.predict(noise)
52
  generated_images = (generated_images * 127.5 + 127.5).astype(np.uint8)
53
  return generated_images.reshape((num_images, 28, 28))
@@ -70,4 +69,4 @@ iface = gr.Interface(
70
  description="Enter a prompt and generate images based on the prompt using GAN.",
71
  example="smiley face"
72
  )
73
- iface.launch()
 
46
  # Функция для генерации изображений на основе запроса
47
  def generate_images(prompt):
48
  num_images = 10
49
+ noise = np.random.normal(0, 1, (num_images, 100))
 
50
  generated_images = generator.predict(noise)
51
  generated_images = (generated_images * 127.5 + 127.5).astype(np.uint8)
52
  return generated_images.reshape((num_images, 28, 28))
 
69
  description="Enter a prompt and generate images based on the prompt using GAN.",
70
  example="smiley face"
71
  )
72
+ iface.launch()