Update app.py

app.py

@@ -10,8 +10,6 @@ from share_btn import community_icon_html, loading_icon_html, share_js
 model_id = "runwayml/stable-diffusion-v1-5"
 device = "cuda"
 
-word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt", use_auth_token=True)
-word_list = word_list_dataset["train"]['text']
 
 is_gpu_busy = False
 def infer(prompt):
@@ -19,10 +17,7 @@ def infer(prompt):
     samples = 4
     steps = 50
     scale = 7.5
-    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 = []
     url = os.getenv('JAX_BACKEND_URL')
     payload = {'prompt': prompt}