Update app.py

app.py

@@ -12,8 +12,13 @@ API_URL = "https://api-inference.huggingface.co/models/openskyml/open-diffusion-
 API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
+word_list_dataset = load_dataset("openskyml/bad-words-prompt-list", data_files="en.txt", use_auth_token=True)
+word_list = word_list_dataset["train"]['text']
 
 def query(prompt, is_negative=False, image_style="None style", steps=8, cfg_scale=7, seed=None, num_images=4):
+    for filter in word_list:
+        if re.search(rf"\b{filter}\b", prompt):
+            raise gr.Error("Unsafe content found. Please try again with different prompts.")
     images = []
     for _ in range(num_images):
         if image_style == "None style":
@@ -51,14 +56,6 @@ def query(prompt, is_negative=False, image_style="None style", steps=8, cfg_scal
 
         image_bytes = requests.post(API_URL, headers=headers, json=payload).content
         image = Image.open(io.BytesIO(image_bytes))
-        rand_num = random.randint(1, 2147483647)
-        api = HfApi()
-        api.upload_file(
-            path_or_fileobj=image,
-            path_in_repo=f"images/img_{rand_num}.png",
-            repo_id="openskyml/prompts",
-            repo_type="dataset",
-        )
         
         images.append(image)