Spaces:
Runtime error
Runtime error
Commit
·
d671906
1
Parent(s):
bace055
Update app.py
Browse files
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 |
-
|
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()
|