Update app.py

app.py

@@ -62,7 +62,7 @@ def download_file(url, folder_path, filename):
 # Download ESRGAN models
 download_file("https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealESRGAN_x2.pth?download=true", "models/upscalers/", "RealESRGAN_x2.pth")
 download_file("https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealESRGAN_x4.pth?download=true", "models/upscalers/", "RealESRGAN_x4.pth")
-download_file("https://huggingface.co/silveroxides/sdxl_lowstep_beta_v1/resolve/main/pytorch_lora_weights.safetensors?download=true", "models/lora/", "LowStepBeta_v001.safetensors")
+# download_file("https://huggingface.co/silveroxides/sdxl_lowstep_beta_v1/resolve/main/pytorch_lora_weights.safetensors?download=true", "models/lora/", "LowStepBeta_v001.safetensors")
 # Download the model files
 ckpt_dir_realpony = snapshot_download(repo_id="silveroxides/RNS_RealPonyV20")
 ckpt_dir_noobai = snapshot_download(repo_id="silveroxides/NoobAI-XL-EPS-1.0-Vwe")
@@ -126,8 +126,8 @@ DEFAULT_NEGATIVE_SUFFIX = "low quality, low resolution, simple background, bad c
 # Initialize Florence model
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-lora_model_path = "models/lora/LowStepBeta_v001.safetensors"
-lora = load_file(lora_model_path, device=device)
+# lora_model_path = "models/lora/LowStepBeta_v001.safetensors"
+# lora = load_file(lora_model_path, device=device)
 
 #def load_models():
     #with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports):
@@ -227,7 +227,7 @@ def generate_image(model_choice, additional_positive_prompt, additional_negative
         pipe = pipe_hybridpony
 
     if use_lowstep_lora:
-        pipe.unet.load_attn_procs(lora)
+        pipeline.unet.load_attn_procs("silveroxides/sdxl_lowstep_beta_v1", weight_name="pytorch_lora_weights.safetensors")
 
     if use_random_seed:
         seed = random.randint(0, 2**32 - 1)