ironjr commited on
Commit
8d7caac
·
verified ·
1 Parent(s): 097b15f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -501,6 +501,8 @@ def draw(state, drawpad):
501
 
502
  root = pathlib.Path(__file__).parent
503
  example_root = os.path.join(root, 'examples')
 
 
504
 
505
  with open(os.path.join(example_root, 'prompt_background.txt')) as f:
506
  prompts_background = [l.strip() for l in f.readlines() if l.strip() != '']
 
501
 
502
  root = pathlib.Path(__file__).parent
503
  example_root = os.path.join(root, 'examples')
504
+ example_images = glob.glob(os.path.join(example_root, '*.png'))
505
+ example_images = [Image.open(i) for i in example_images]
506
 
507
  with open(os.path.join(example_root, 'prompt_background.txt')) as f:
508
  prompts_background = [l.strip() for l in f.readlines() if l.strip() != '']