sasha HF Staff commited on
Commit
2156ab9
·
verified ·
1 Parent(s): 3af88b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,7 +7,7 @@ import tempfile
7
  def open_sd_ims(adj, group, seed):
8
  if group != '':
9
  if adj != '':
10
- prompt=adj+'_'+group.replace(' ','_')
11
  if os.path.isdir(prompt) == False:
12
  shutil.unpack_archive('zipped_images/stablediffusion/'+ prompt.replace(' ', '_') +'.zip', prompt, 'zip')
13
  else:
@@ -25,7 +25,7 @@ def open_ims(model, adj, group):
25
  if model == "Dall-E 2":
26
  if group != '':
27
  if adj != '':
28
- prompt=adj+'_'+group.replace(' ','_')
29
  if os.path.isdir(tmpdirname + '/' + model.replace(' ','').lower()+ '/'+ prompt) == False:
30
  shutil.unpack_archive('zipped_images/'+ model.replace(' ','').lower()+ '/'+ prompt.replace(' ', '_') +'.zip', tmpdirname+ '/'+ model.replace(' ','').lower()+ '/'+ prompt, 'zip')
31
  else:
@@ -39,7 +39,7 @@ def open_ims(model, adj, group):
39
  else:
40
  if group != '':
41
  if adj != '':
42
- prompt=adj+'_'+group.replace(' ','_')
43
  if os.path.isdir(tmpdirname + '/' + model.replace(' ','').lower()+ '/'+ prompt) == False:
44
  shutil.unpack_archive('zipped_images/'+ model.replace(' ','').lower()+ '/'+ prompt.replace(' ', '_') +'.zip', tmpdirname + '/' +model.replace(' ','').lower()+ '/'+ prompt, 'zip')
45
  else:
 
7
  def open_sd_ims(adj, group, seed):
8
  if group != '':
9
  if adj != '':
10
+ prompt=adj+'_'+str(group).replace(' ','_')
11
  if os.path.isdir(prompt) == False:
12
  shutil.unpack_archive('zipped_images/stablediffusion/'+ prompt.replace(' ', '_') +'.zip', prompt, 'zip')
13
  else:
 
25
  if model == "Dall-E 2":
26
  if group != '':
27
  if adj != '':
28
+ prompt=adj+'_'+str(group).replace(' ','_')
29
  if os.path.isdir(tmpdirname + '/' + model.replace(' ','').lower()+ '/'+ prompt) == False:
30
  shutil.unpack_archive('zipped_images/'+ model.replace(' ','').lower()+ '/'+ prompt.replace(' ', '_') +'.zip', tmpdirname+ '/'+ model.replace(' ','').lower()+ '/'+ prompt, 'zip')
31
  else:
 
39
  else:
40
  if group != '':
41
  if adj != '':
42
+ prompt=adj+'_'+str(group).replace(' ','_')
43
  if os.path.isdir(tmpdirname + '/' + model.replace(' ','').lower()+ '/'+ prompt) == False:
44
  shutil.unpack_archive('zipped_images/'+ model.replace(' ','').lower()+ '/'+ prompt.replace(' ', '_') +'.zip', tmpdirname + '/' +model.replace(' ','').lower()+ '/'+ prompt, 'zip')
45
  else: