Add some prompts for even more diversity

app.py

@@ -35,6 +35,30 @@ safe_prompts = [
     "a person, black sweater, long coat, gray trousers"
 ]
 
+additional_safe_prompts = [
+    "a person, white blouse, navy skirt, black flats",
+    "a person, denim overalls, striped shirt, sneakers",
+    "a person, green parka, wool scarf, winter boots",
+    "a person, silk shirt, high-waisted trousers, loafers",
+    "a person, traditional sari, gold bangles, sandals",
+    "a person, lab coat, dress shirt, slacks",
+    "a person, chef uniform, apron, white hat",
+    "a person, flight attendant uniform, scarf, heels",
+    "a person, construction vest, hard hat, work boots",
+    "a person, navy sweater, corduroy pants, brown boots",
+    "a person, red hoodie, jeans, running shoes",
+    "a person, black tunic, leggings, sandals",
+    "a person, plaid shirt, jeans, cowboy boots",
+    "a person, white jumpsuit, belt, ankle boots",
+    "a person, vintage dress, pearl earrings, pumps",
+    "a person, medical scrubs, sneakers, ID badge",
+    "a person, blue windbreaker, track pants, sneakers",
+    "a person, bohemian blouse, long skirt, sandals",
+    "a person, blazer, graphic tee, ripped jeans",
+    "a person, puffer jacket, beanie, hiking shoes"
+]
+
+safe_prompts += additional_safe_prompts
 
 pipe_safety = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")