Update app.py

app.py

@@ -117,7 +117,7 @@ license_str_en_orig = """
 """
 
 @spaces.GPU
-def infer(
+def infer_impl(
     prompt,
     style,
     lang='ja',
@@ -132,7 +132,6 @@ def infer(
     progress=gr.Progress(track_tqdm=True),
 ):
     now = datetime.datetime.now()
-    print("button clickeds : ", now, prompt, style, lang)
     style_to_prompt  = {
         "sensei art / 先生アート" : "先生アートsensei artイラスト", 
         "digital illustration / デジタルイラスト" : "デジタルイラスト、digital illustration", 
@@ -163,8 +162,6 @@ def infer(
         prompt = style_to_prompt[style] + ", " + prompt
         # prompt = prompt  + " " + style_to_prompt[style]
     negative_prompt = negative_prompt + ", " + style2negative_prompt[style]
-    print("prompt : ", prompt)
-    print("ne_prompt : ", negative_prompt)
     
     width, height = 768, 768
     if ar == '16:9':
@@ -241,6 +238,24 @@ def infer(
     image_w_logo.paste(likes_logo_img, (0, image.height))
     yield image_w_logo, license_str, detected_public_fictional_characters_info, seed
 
+def infer(
+    prompt,
+    style,
+    lang='ja',
+    negative_prompt="elan doodle",
+    seed=42,
+    randomize_seed=True,
+    ar="1:1",
+    # width=672,
+    # height=896,
+    guidance_scale=5.0,
+    num_inference_steps=40,
+    progress=gr.Progress(track_tqdm=True),
+):
+    if any([a in prompt for a in bad_words]):
+        return None, "## <span style='color:orangered'>ERROR: Invalid prompt / 不適切なプロンプト </span>", None, seed    
+    return infer_impl(prompt, style, lang, negative_prompt, seed, randomize_seed, ar, guidance_scale, num_inference_steps, progress)
+
 
 # Seafoam theme based on
 # https://huggingface.co/spaces/gradio/seafoam