Spaces:
Running
on
L40S
Running
on
L40S
Adding even more safe prompt
Browse files
app.py
CHANGED
|
@@ -58,7 +58,42 @@ additional_safe_prompts = [
|
|
| 58 |
"a person, puffer jacket, beanie, hiking shoes"
|
| 59 |
]
|
| 60 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 61 |
safe_prompts += additional_safe_prompts
|
|
|
|
|
|
|
| 62 |
|
| 63 |
pipe_safety = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")
|
| 64 |
|
|
|
|
| 58 |
"a person, puffer jacket, beanie, hiking shoes"
|
| 59 |
]
|
| 60 |
|
| 61 |
+
more_safe_prompts = [
|
| 62 |
+
"a person, olive green jumpsuit, white sneakers, aviator sunglasses",
|
| 63 |
+
"a person, oversized sweater, leggings, knee-high boots",
|
| 64 |
+
"a person, red blazer, white blouse, black trousers",
|
| 65 |
+
"a person, short-sleeve shirt, cargo pants, hiking boots",
|
| 66 |
+
"a person, pinstripe suit, maroon tie, dress shoes",
|
| 67 |
+
"a person, checkered dress, denim jacket, ankle boots",
|
| 68 |
+
"a person, hooded parka, snow pants, winter gloves",
|
| 69 |
+
"a person, silk scarf, long coat, leather gloves",
|
| 70 |
+
"a person, floral print shirt, beige trousers, sandals",
|
| 71 |
+
"a person, sports jersey, athletic shorts, running shoes",
|
| 72 |
+
"a person, fisherman sweater, beanie, waterproof boots",
|
| 73 |
+
"a person, vest, flannel shirt, hiking pants",
|
| 74 |
+
"a person, polo shirt, chinos, loafers",
|
| 75 |
+
"a person, elegant gown, clutch bag, heels",
|
| 76 |
+
"a person, lightweight jacket, white jeans, sneakers",
|
| 77 |
+
"a person, black hoodie, cargo pants, sneakers",
|
| 78 |
+
"a person, striped sweater, denim skirt, tights",
|
| 79 |
+
"a person, leather trench coat, turtleneck, boots",
|
| 80 |
+
"a person, yellow sundress, straw hat, sandals",
|
| 81 |
+
"a person, traditional hanbok, embroidered top, ribbon belt",
|
| 82 |
+
"a person, navy cardigan, floral blouse, jeans",
|
| 83 |
+
"a person, formal coat, scarf, dress shoes",
|
| 84 |
+
"a person, sweater vest, dress shirt, pleated pants",
|
| 85 |
+
"a person, short-sleeved dress, flats, handbag",
|
| 86 |
+
"a person, zip-up fleece, track pants, sneakers",
|
| 87 |
+
"a person, apron, collared shirt, khakis",
|
| 88 |
+
"a person, cultural dashiki, fitted pants, sandals",
|
| 89 |
+
"a person, business jacket, turtleneck, slacks",
|
| 90 |
+
"a person, rain poncho, hiking pants, waterproof shoes",
|
| 91 |
+
"a person, dark red blazer, black shirt, brown pants"
|
| 92 |
+
]
|
| 93 |
+
|
| 94 |
safe_prompts += additional_safe_prompts
|
| 95 |
+
safe_prompts += more_safe_prompts
|
| 96 |
+
|
| 97 |
|
| 98 |
pipe_safety = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")
|
| 99 |
|