yaron123 commited on
Commit
aca79b5
·
1 Parent(s): 0d610cf
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -613,7 +613,6 @@ def all_pipes(pos,neg,artist,song):
613
 
614
  imgs = pipe_generate_image(pos,neg)
615
 
616
- index = 1
617
  for i in range(len(imgs)):
618
  imgs[i] = upscaler(imgs[i])
619
 
@@ -644,12 +643,14 @@ def handle_generation(artist,song,genre,lyrics):
644
 
645
  imgs = all_pipes(pos,neg,pos_artist,pos_song)
646
 
 
647
  names = []
648
  for img in imgs:
649
  labeled_img = add_song_cover_text(img,artist,song,height*4,width*4)
650
  name = f'{artist} - {song} ({index}).png'
651
  labeled_img.save(name)
652
  names.append(name)
 
653
 
654
  return names
655
 
 
613
 
614
  imgs = pipe_generate_image(pos,neg)
615
 
 
616
  for i in range(len(imgs)):
617
  imgs[i] = upscaler(imgs[i])
618
 
 
643
 
644
  imgs = all_pipes(pos,neg,pos_artist,pos_song)
645
 
646
+ index = 1
647
  names = []
648
  for img in imgs:
649
  labeled_img = add_song_cover_text(img,artist,song,height*4,width*4)
650
  name = f'{artist} - {song} ({index}).png'
651
  labeled_img.save(name)
652
  names.append(name)
653
+ index = index + 1
654
 
655
  return names
656