Update app.py

app.py

@@ -3,13 +3,14 @@
 # gr.Interface.load("models/Akbartus/Lora360").launch(show_api=True)
 
 import gradio as gr
-from diffusers import StableDiffusionPipeline
+from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 
 def process_image(prompt, num_inference_steps, guidance_scale):
-    model_path = "Akbartus/Lora360"
-    pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_safetensors=True)
-    pipe.unet.load_attn_procs(model_path)
+
+    pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", safety_checker=None, use_safetensors=True)
     pipe.to("cpu")
+    pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, use_karras_sigmas=True)
+    pipe.load_lora_weights("Akbartus/Lora360", weight_name="360Diffusion_v1.safetensors")
     image = pipe(prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale).images[0]
     return image