WiNE-iNEFF commited on
Commit
7c04462
1 Parent(s): d0f0ae5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def promptgen(choice, num, artist):
20
  vocab = len(prompt)
21
  generated = []
22
  artists_num = 0
23
- while len(sorted(set(generated), key=lambda d: generated.index(d))) < num:
24
  rand = random.choice(prompt)
25
  if rand.startswith('art by') and artists_num < artist:
26
  artists_num +=1
 
20
  vocab = len(prompt)
21
  generated = []
22
  artists_num = 0
23
+ while len(sorted(set(generated), key=lambda d: generated.index(d))) < int(num):
24
  rand = random.choice(prompt)
25
  if rand.startswith('art by') and artists_num < artist:
26
  artists_num +=1