Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -144,8 +144,10 @@ def main():
|
|
144 |
unsafe_allow_html=True,)
|
145 |
|
146 |
if generate_button:
|
147 |
-
|
148 |
-
|
|
|
|
|
149 |
with col12:
|
150 |
st.image(gan_images[0], width=300)
|
151 |
if len(gan_images) > 1:
|
|
|
144 |
unsafe_allow_html=True,)
|
145 |
|
146 |
if generate_button:
|
147 |
+
with st.spinner(text=f"Loading selected model..."):
|
148 |
+
generator = load_generator(model_name[img_type])
|
149 |
+
with st.spinner(text=f"Generating images..."):
|
150 |
+
gan_images = generate_images(generator, number_imgs)
|
151 |
with col12:
|
152 |
st.image(gan_images[0], width=300)
|
153 |
if len(gan_images) > 1:
|