Update app.py

app.py

@@ -6,7 +6,6 @@ from peft import PeftModel, LoraConfig
 import os
 
 def get_lora_sd_pipeline(
-    #ckpt_dir='./lora_logos', 
     ckpt_dir='./lora_man_animestyle',
     base_model_name_or_path=None, 
     dtype=torch.float16, 
@@ -54,11 +53,9 @@ def align_embeddings(prompt_embeds, negative_prompt_embeds):
            torch.nn.functional.pad(negative_prompt_embeds, (0, 0, 0, max_length - negative_prompt_embeds.shape[1]))
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-#model_id_default = "CompVis/stable-diffusion-v1-4"
 model_id_default = "stable-diffusion-v1-5/stable-diffusion-v1-5"
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
-#pipe_default = get_lora_sd_pipeline(ckpt_dir='./lora_logos', base_model_name_or_path=model_id_default, dtype=torch_dtype).to(device)
 pipe_default = get_lora_sd_pipeline(ckpt_dir='./lora_man_animestyle', base_model_name_or_path=model_id_default, dtype=torch_dtype).to(device)
 
 
@@ -71,7 +68,6 @@ def infer(
     width=512, 
     height=512, 
     num_inference_steps=20, 
-    #model_id='CompVis/stable-diffusion-v1-4',
     model_id='stable-diffusion-v1-5/stable-diffusion-v1-5', 
     seed=42, 
     guidance_scale=7.0, 
@@ -120,9 +116,8 @@ with gr.Blocks(css=css) as demo:
         
         with gr.Row():
             model_id = gr.Textbox(
-                label="Model ID",
+                label="Model",
                 max_lines=1,
-                #placeholder="Enter model id like 'CompVis/stable-diffusion-v1-4'",
                 placeholder="Enter model id like 'stable-diffusion-v1-5/stable-diffusion-v1-5'",
                 value=model_id_default
             )
@@ -140,13 +135,13 @@ with gr.Blocks(css=css) as demo:
         )
 
         with gr.Row():
-            seed = gr.Number(
-                label="Seed",
-                minimum=0,
-                maximum=MAX_SEED,
-                step=1,
-                value=42,
-            )
+            lora_scale = gr.Slider(
+                label="LoRA scale",
+                minimum=0.0,
+                maximum=1.0,
+                step=0.1,
+                value=0.5,
+            ) 
 
         with gr.Row():
             guidance_scale = gr.Slider(
@@ -154,16 +149,16 @@ with gr.Blocks(css=css) as demo:
                 minimum=0.0,
                 maximum=10.0,
                 step=0.1,
-                value=7.0,
-            )
-
+                value=7.5,
+            )  
+        
         with gr.Row():
-            lora_scale = gr.Slider(
-                label="LoRA scale",
-                minimum=0.0,
-                maximum=1.0,
-                step=0.1,
-                value=0.5,
+            seed = gr.Number(
+                label="Seed",
+                minimum=0,
+                maximum=MAX_SEED,
+                step=1,
+                value=42,
             )
 
         with gr.Row():