Update app.py

app.py

@@ -53,6 +53,19 @@ def load_generated_images():
     image_files.sort(key=lambda x: os.path.getctime(x), reverse=True)
     return image_files
 
+def load_predefined_images():
+    # You'll need to create an 'assets' directory with your sample images
+    # Return empty list if assets directory or images don't exist
+    if not os.path.exists('assets'):
+        return []
+    
+    predefined_images = [
+        os.path.join('assets', f) 
+        for f in os.listdir('assets') 
+        if f.endswith(('.png', '.jpg', '.jpeg', '.webp'))
+    ]
+    return sorted(predefined_images)[:6]  # Limit to 6 images
+
 @spaces.GPU(duration=120)
 def inference(
     prompt: str,