MWAI-HF_Simple_Prompt_Generator

Runtime error

WiNE-iNEFF commited on Feb 18, 2023

Commit

a573473

1 Parent(s): 6d56b12

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ def promptgen(choice, num, artist):
     rand = random.randint(0, vocab)
     if prompt[rand-1].startswith('art by') and artists_num < artist:
         artists_num +=1
-		generated.append(prompt[rand-1])
 	elif not prompt[rand-1].startswith('art by'):
 		generated.append(prompt[rand-1])
   print(' '.join(set(generated)) + '\n')

     rand = random.randint(0, vocab)
     if prompt[rand-1].startswith('art by') and artists_num < artist:
         artists_num +=1
+        generated.append(prompt[rand-1])
 	elif not prompt[rand-1].startswith('art by'):
 		generated.append(prompt[rand-1])
   print(' '.join(set(generated)) + '\n')