salomonsky commited on
Commit
c6afe27
1 Parent(s): 5a497fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -102,6 +102,9 @@ def gen(prompts, width, height, model_name, num_variants=1, use_enhanced=True):
102
  st.error(f"Error al generar imágenes: {e}")
103
  return images
104
 
 
 
 
105
  def display_gallery():
106
  st.header("Galería de Imágenes Guardadas")
107
  images = list_saved_images()
 
102
  st.error(f"Error al generar imágenes: {e}")
103
  return images
104
 
105
+ def list_saved_images():
106
+ return list(DATA_PATH.glob("*.jpg"))
107
+
108
  def display_gallery():
109
  st.header("Galería de Imágenes Guardadas")
110
  images = list_saved_images()