WiNE-iNEFF
commited on
Update app.py
Browse files
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
|