Update app.py
Browse files
app.py
CHANGED
@@ -18,6 +18,7 @@ import requests
|
|
18 |
import torch
|
19 |
from diffusers import DiffusionPipeline
|
20 |
import importlib
|
|
|
21 |
from urllib.parse import urlparse
|
22 |
|
23 |
random.seed(time.time())
|
@@ -28,7 +29,24 @@ USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
|
|
28 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
|
29 |
|
30 |
NUM_IMAGES_PER_PROMPT = 1
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
31 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
32 |
|
33 |
cfg = json.load(open("app.conf"))
|
34 |
|
@@ -254,10 +272,10 @@ def generate(p, progress=gr.Progress(track_tqdm=True)):
|
|
254 |
inference_steps = cfg.get("inference_steps", 30)
|
255 |
randomize_seed = True
|
256 |
guidance_scale = cfg.get("guidance_scale", 7.5)
|
|
|
257 |
prompt_str = cfg.get("prompt", "{prompt}").replace("{prompt}", p)
|
258 |
seed = int(randomize_seed_fn(seed, randomize_seed))
|
259 |
generator = torch.Generator(pipe.device).manual_seed(seed)
|
260 |
-
|
261 |
images = pipe(
|
262 |
prompt=prompt_str,
|
263 |
negative_prompt=negative_prompt,
|
|
|
18 |
import torch
|
19 |
from diffusers import DiffusionPipeline
|
20 |
import importlib
|
21 |
+
import re
|
22 |
from urllib.parse import urlparse
|
23 |
|
24 |
random.seed(time.time())
|
|
|
29 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
|
30 |
|
31 |
NUM_IMAGES_PER_PROMPT = 1
|
32 |
+
# Define the regular expression
|
33 |
+
child_related_regex = re.compile(
|
34 |
+
r'(child|children|kid|kids|baby|babies|toddler|infant|juvenile|minor|underage|preteen|adolescent|youngster|youth|kindergarten|preschool|'
|
35 |
+
r'([1-9]|1[0-7])\s*year(s)?\s*old|' # Matches 1 to 17 years old
|
36 |
+
r'little|small|tiny|short|new\s*born\s*(boy|girl|bro|brother|sis|sister))',
|
37 |
+
re.IGNORECASE
|
38 |
+
)
|
39 |
|
40 |
+
# Function to remove child-related content from a prompt
|
41 |
+
def remove_child_related_content(prompt):
|
42 |
+
cleaned_prompt = re.sub(child_related_regex, '', prompt)
|
43 |
+
return cleaned_prompt.strip()
|
44 |
+
|
45 |
+
# Function to check if a prompt contains child-related content
|
46 |
+
def contains_child_related_content(prompt):
|
47 |
+
if child_related_regex.search(prompt):
|
48 |
+
return True
|
49 |
+
return False
|
50 |
|
51 |
cfg = json.load(open("app.conf"))
|
52 |
|
|
|
272 |
inference_steps = cfg.get("inference_steps", 30)
|
273 |
randomize_seed = True
|
274 |
guidance_scale = cfg.get("guidance_scale", 7.5)
|
275 |
+
p = remove_child_related_content(p)
|
276 |
prompt_str = cfg.get("prompt", "{prompt}").replace("{prompt}", p)
|
277 |
seed = int(randomize_seed_fn(seed, randomize_seed))
|
278 |
generator = torch.Generator(pipe.device).manual_seed(seed)
|
|
|
279 |
images = pipe(
|
280 |
prompt=prompt_str,
|
281 |
negative_prompt=negative_prompt,
|