Update app.py

app.py

@@ -37,11 +37,11 @@ def error_str(error, title="Error"):
             {error}"""  if error else ""
 
 
-def inference(prompt, guidance, steps, image_size="Square", seed=0, img=None, strength=0.5, neg_prompt="", cool_japan_type="Anime", disable_auto_prompt_correction=False):
+def inference(prompt, guidance, steps, image_size="Square", seed=0, img=None, strength=0.5, neg_prompt="", disable_auto_prompt_correction=False):
 
   generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
 
-  prompt,neg_prompt=auto_prompt_correction(prompt,neg_prompt,cool_japan_type,disable_auto_prompt_correction)
+  prompt,neg_prompt=auto_prompt_correction(prompt,neg_prompt,disable_auto_prompt_correction)
 
   if(image_size=="Portrait"):
       height=1024
@@ -67,8 +67,6 @@ def inference(prompt, guidance, steps, image_size="Square", seed=0, img=None, st
     return None, error_str(e)
 def auto_prompt_correction(prompt_ui,neg_prompt_ui,disable_auto_prompt_correction):
     # auto prompt correction
-
-        
     prompt=str(prompt_ui)
     neg_prompt=str(neg_prompt_ui)
     prompt=prompt.lower()
@@ -85,13 +83,13 @@ def auto_prompt_correction(prompt_ui,neg_prompt_ui,disable_auto_prompt_correctio
     splited_prompt=prompt.replace(","," ").replace("_"," ").split(" ")
     for word in human_words:
         if( word in splited_prompt):
-            prompt=f"{cool_japan_type}, masterpiece, {prompt}, good pupil, 4k, detailed"
+            prompt=f"anime, masterpiece, {prompt}, good pupil, 4k, detailed"
             neg_prompt=f"(((deformed))), blurry, ((((bad anatomy)))), {neg_prompt}, bad pupil, disfigured, poorly drawn face, mutation, mutated, (extra limb), (ugly), (poorly drawn hands), bad hands, fused fingers, messy drawing, broken legs censor, low quality, (mutated hands and fingers:1.5), (long body :1.3), (mutation, poorly drawn :1.2), ((bad eyes)), ui, error, missing fingers, fused fingers, one hand with more than 5 fingers, one hand with less than 5 fingers, one hand with more than 5 digit, one hand with less than 5 digit, extra digit, fewer digits, fused digit, missing digit, bad digit, liquid digit, long body, uncoordinated body, unnatural body, lowres, jpeg artifacts, 3d, cg, text"
 
     animal_words=["cat","dog","bird"]
     for word in animal_words:
         if( word in splited_prompt):
-            prompt=f"{cool_japan_type}, a {word}, 4k, detailed"
+            prompt=f"anime, a {word}, 4k, detailed"
             neg_prompt=f"(((deformed))), blurry, ((((bad anatomy)))), {neg_prompt}, bad pupil, disfigured, poorly drawn face, mutation, mutated, (extra limb), (ugly), (poorly drawn hands), bad hands, fused fingers, messy drawing, broken legs censor, low quality, (mutated hands and fingers:1.5), (long body :1.3), (mutation, poorly drawn :1.2), ((bad eyes)), ui, error, missing fingers, fused fingers, one hand with more than 5 fingers, one hand with less than 5 fingers, one hand with more than 5 digit, one hand with less than 5 digit, extra digit, fewer digits, fused digit, missing digit, bad digit, liquid digit, long body, uncoordinated body, unnatural body, lowres, jpeg artifacts, 3d, cg, text"
 
     background_words=["mount fuji","mt. fuji","building", "buildings", "tokyo", "kyoto", "nara", "shibuya", "shinjuku"]
@@ -143,7 +141,7 @@ with gr.Blocks(css=css) as demo:
                Demo for <a href="https://huggingface.co/aipicasso/picasso-diffusion-1-0">Picasso Diffusion 1.0</a> .<br>
               </p>
               <p>
-              サンプル： そのままGenerateボタンを押してください。
+              サンプル： そのままGenerateボタンを押してください。<br>
               sample : Click "Generate" button without any prompts.
               </p>
               <p>
@@ -153,7 +151,7 @@ with gr.Blocks(css=css) as demo:
               sample prompt2 : boy, school uniform
               </p>
               Running on {"<b>GPU 🔥</b>" if torch.cuda.is_available() else f"<b>CPU 🥶</b>. For faster inference it is recommended to <b>upgrade to GPU in <a href='https://huggingface.co/spaces/akhaliq/cool-japan-diffusion-2-1-0/settings'>Settings</a></b>"} <br>
-              <a style="display:inline-block" href="https://huggingface.co/spaces/aipicasso/cool-japan-diffusion-latest-demo?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a> to say goodbye from waiting for the generating.
+              <a style="display:inline-block" href="https://huggingface.co/spaces/aipicasso/picasso-diffusion-latest-demo?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a> to say goodbye from waiting for the generating.
             </div>
         """
     )
@@ -161,11 +159,6 @@ with gr.Blocks(css=css) as demo:
         
         with gr.Column(scale=55):
           with gr.Group():
-              with gr.Row():
-                cool_japan_type=gr.Radio(["Anime", "Manga", "Game"])
-                cool_japan_type.show_label=False
-                cool_japan_type.value="Anime"
-                
               with gr.Row():
                 prompt = gr.Textbox(label="Prompt", show_label=False, max_lines=2,placeholder="[your prompt]").style(container=False)
                 generate = gr.Button(value="Generate").style(rounded=(False, True, True, False))
@@ -194,7 +187,7 @@ with gr.Blocks(css=css) as demo:
                 image = gr.Image(label="Image", height=256, tool="editor", type="pil")
                 strength = gr.Slider(label="Transformation strength", minimum=0, maximum=1, step=0.01, value=0.5)
                   
-    inputs = [prompt, guidance, steps, image_size, seed, image, strength, neg_prompt, cool_japan_type, disable_auto_prompt_correction]
+    inputs = [prompt, guidance, steps, image_size, seed, image, strength, neg_prompt, disable_auto_prompt_correction]
 
     outputs = [image_out, error_output]
     prompt.submit(inference, inputs=inputs, outputs=outputs)