Update app.py

app.py

@@ -14,7 +14,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 
 # Load the diffusion pipeline
 pipe = StableDiffusionXLPipeline.from_single_file(
-    "https://huggingface.co/kayfahaarukku/AkashicPulse-v1.0/blob/main/AkashicPulse-v1.0-ft-ft.safetensors",  # Fixed URL
+    "https://huggingface.co/kayfahaarukku/AkashicPulse-v1.0/blob/main/AkashicPulse-v1.0-ft-ft.safetensors",
     torch_dtype=torch.float16,
     custom_pipeline="lpw_stable_diffusion_xl",
     use_safetensors=True,
@@ -23,16 +23,16 @@ pipe = StableDiffusionXLPipeline.from_single_file(
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
 # Function to generate an image
-@spaces.GPU  # Adjust the duration as needed
+@spaces.GPU
 def generate_image(prompt, negative_prompt, use_defaults, resolution, guidance_scale, num_inference_steps, seed, randomize_seed, progress=gr.Progress()):
     try:
-        pipe.to('cuda')  # Move the model to GPU when the function is called
+        pipe.to('cuda')
         
         if randomize_seed:
             seed = random.randint(0, 99999999)
         if use_defaults:
-            prompt = f"{prompt}, best quality, amazing quality, very aesthetic"
-            negative_prompt = f"nsfw, lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract], {negative_prompt}"
+            prompt = f"{prompt}, masterpiece, best quality"
+            negative_prompt = f"lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, signature, watermark, username, blurry, {negative_prompt}"
         generator = torch.manual_seed(seed)
         
         def callback(step, timestep, latents):
@@ -68,11 +68,11 @@ def interface_fn(prompt, negative_prompt, use_defaults, resolution, guidance_sca
     return image, seed, gr.update(value=metadata_text)
 
 def reset_inputs():
-    return gr.update(value=''), gr.update(value=''), gr.update(value=True), gr.update(value='832x1216'), gr.update(value=4), gr.update(value=28), gr.update(value=0), gr.update(value=True), gr.update(value='')
+    return gr.update(value=''), gr.update(value=''), gr.update(value=True), gr.update(value='832x1216'), gr.update(value=7), gr.update(value=28), gr.update(value=0), gr.update(value=True), gr.update(value='')
 
-with gr.Blocks(title="irAsu 1.0 Demo", theme="NoCrypt/[email protected]") as demo:
+with gr.Blocks(title="AkashicPulse Demo", theme="NoCrypt/[email protected]") as demo:
     gr.HTML(
-        "<h1>irAsu 1.0 Demo</h1>"
+        "<h1>AkashicPulse Demo</h1>"
         "<p>This demo is intended to showcase what the model is capable of and is not intended to be the main generation platform. "
         "Results produced with Diffusers are not the best, and it's highly recommended for you to get the model running inside "
         "Stable Diffusion WebUI or ComfyUI.</p>"
@@ -90,7 +90,7 @@ with gr.Blocks(title="irAsu 1.0 Demo", theme="NoCrypt/[email protected]") as demo:
                 label="Resolution",
                 value="832x1216"
             )
-            guidance_scale_input = gr.Slider(minimum=1, maximum=20, step=0.5, label="Guidance Scale", value=4)
+            guidance_scale_input = gr.Slider(minimum=1, maximum=20, step=0.5, label="Guidance Scale", value=7)
             num_inference_steps_input = gr.Slider(minimum=1, maximum=100, step=1, label="Number of Inference Steps", value=28)
             seed_input = gr.Slider(minimum=0, maximum=999999999, step=1, label="Seed", value=0, interactive=True)
             randomize_seed_input = gr.Checkbox(label="Randomize Seed", value=True)
@@ -104,15 +104,15 @@ with gr.Blocks(title="irAsu 1.0 Demo", theme="NoCrypt/[email protected]") as demo:
                 metadata_textbox = gr.Textbox(lines=6, label="Image Parameters", interactive=False, max_lines=6)
             gr.Markdown(
                 """
-                ### Recommended prompt formatting:
-                `1girl/1boy, character name, from what series, everything else in any order, best quality, amazing quality, very aesthetic,`
-
-                **PS:** `best quality, amazing quality, very aesthetic,` is automatically added when "Use Default Quality Tags and Negative Prompt" is enabled
-
                 ### Recommended settings:
-                - Steps: 25-30
-                - CFG: 3.5-5
-                - Sweet spot: 28 steps, 4 CFG
+                - Sampling: Euler a
+                - Steps: 20-30 (sweet spot: 28)
+                - CFG: 4-10 (sweet spot: 7)
+                - [Not mandatory] On reForge or ComfyUI, have MaHiRo CFG enabled
+
+                ### Recommended prompt formatting:
+                - Prompt: [1girl/1boy], [character name], [series], by [artist name], [the rest of the prompt], masterpiece, best quality
+                - Negative prompt: lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, signature, watermark, username, blurry, [the rest of the negative prompt]
                 """
             )