Update app.py

app.py

@@ -19,6 +19,7 @@ import gradio as gr
 from PIL import Image
 import numpy as np
 from transformers import AutoProcessor, AutoModelForCausalLM, pipeline
+from safetensors.torch import load_file
 import requests
 from RealESRGAN import RealESRGAN
 
@@ -62,8 +63,6 @@ def download_file(url, folder_path, filename):
 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")
-lora_model_path = "models/lora/LowStepBeta_v001.safetensors"
-lora = torch.load(lora_model_path)
 # 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")
@@ -104,6 +103,9 @@ pipe_realpony.unet.set_attn_processor(AttnProcessor2_0())
 pipe_noobai.unet.set_attn_processor(AttnProcessor2_0())
 pipe_hybridpony.unet.set_attn_processor(AttnProcessor2_0())
 
+lora_model_path = "models/lora/LowStepBeta_v001.safetensors"
+lora = load_file(lora_model_path).to("cuda")
+
 # Define samplers
 samplers = {
     "Euler a": EulerAncestralDiscreteScheduler.from_config(pipe_realpony.scheduler.config),