Update app.py

app.py

@@ -3,6 +3,14 @@ import os
 os.system("git clone --recursive https://github.com/JD-P/cloob-latent-diffusion")
 os.system("cd cloob-latent-diffusion;pip install omegaconf pillow pytorch-lightning einops wandb ftfy regex ./CLIP")
 
+# setup some example images
+examples = []
+EXAMPLES_DIR = 'examples'
+if os.path.isdir(EXAMPLES_DIR):
+    for file in os.listdir(EXAMPLES_DIR):
+        path = EXAMPLES_DIR + "/" + file
+        examples.append([path, DEFAULT_PROMPT])
+
 import argparse
 from functools import partial
 from pathlib import Path
@@ -35,6 +43,7 @@ ae_config_path = cached_download(hf_hub_url("huggan/ccld_wa", filename="ae_model
 
 # Define a few utility functions
 
+
 def parse_prompt(prompt, default_weight=3.):
     if prompt.startswith('http://') or prompt.startswith('https://'):
         vals = prompt.rsplit(':', 2)