mbrack commited on
Commit
4a70338
1 Parent(s): 6352ff4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -17,13 +17,15 @@ with zipfile.ZipFile("images/stable_diffusion.zip","r") as zip_ref:
17
  def open_stable_ims(profession):
18
  if len(profession) != 0:
19
  dirname = 'images/stable_diffusion/'+ profession+'/'
20
- images = random.shuffle([Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)])
 
21
  return images[:16]
22
 
23
  def open_fair_ims(profession):
24
  if len(profession) != 0:
25
  dirname = 'images/fair_diffusion/' + profession+'/'
26
- images = random.shuffle([Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)])
 
27
  return images[:16]
28
 
29
 
 
17
  def open_stable_ims(profession):
18
  if len(profession) != 0:
19
  dirname = 'images/stable_diffusion/'+ profession+'/'
20
+ images = [Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)]
21
+ random.shuffle(images)
22
  return images[:16]
23
 
24
  def open_fair_ims(profession):
25
  if len(profession) != 0:
26
  dirname = 'images/fair_diffusion/' + profession+'/'
27
+ images = [Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)]
28
+ random.shuffle(images)
29
  return images[:16]
30
 
31