Update app.py

app.py

@@ -439,18 +439,18 @@ def generate(
         if pipe is None:
             raise gr.Error("Model not loaded.")
 
-        has_unsafe_concept = is_unsafe_prompt(
-            pipe.text_encoder,
-            pipe.tokenizer,
-            system_prompt=UNSAFE_PROMPT_CHECK,
-            user_prompt=prompt,
-            max_new_token=UNSAFE_MAX_NEW_TOKEN,
-        )
-        if has_unsafe_concept:
-            raise UnsafeContentError("Input unsafe")
+        # has_unsafe_concept = is_unsafe_prompt(
+        #     pipe.text_encoder,
+        #     pipe.tokenizer,
+        #     system_prompt=UNSAFE_PROMPT_CHECK,
+        #     user_prompt=prompt,
+        #     max_new_token=UNSAFE_MAX_NEW_TOKEN,
+        # )
+        # if has_unsafe_concept:
+        #     raise UnsafeContentError("Input unsafe")
 
         final_prompt = prompt
-
+        enhance = False
         if enhance:
             final_prompt, _ = prompt_enhance(prompt, True)
             print(f"Enhanced prompt: {final_prompt}")