Update app.py

app.py

@@ -5,6 +5,13 @@ from diffusers import StableDiffusionPipeline
 from peft import PeftModel, LoraConfig
 import os
 
+MAX_SEED = np.iinfo(np.int32).max
+MAX_IMAGE_SIZE = 1024
+
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model_id_default = "stable-diffusion-v1-5/stable-diffusion-v1-5"
+torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
+
 def get_lora_sd_pipeline(
     ckpt_dir='./lora_man_animestyle',
     base_model_name_or_path=None, 
@@ -52,16 +59,8 @@ def align_embeddings(prompt_embeds, negative_prompt_embeds):
     return torch.nn.functional.pad(prompt_embeds, (0, 0, 0, max_length - prompt_embeds.shape[1])), \
            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 = "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_man_animestyle', base_model_name_or_path=model_id_default, dtype=torch_dtype).to(device)
 
-
-MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 1024
-
 def infer(
     prompt, 
     negative_prompt,