Update app.py

app.py

@@ -2,10 +2,10 @@ import torch
 from diffusers import StableDiffusionPipeline
 import gradio as gr
 
-model_base = "https://huggingface.co/Krebzonide/LazyMixPlus/blob/main/lazymixRealAmateur_v30b.safetensors"
+model_base = "runwayml/stable-diffusion-v1-5"
 lora_model_path = "Krebzonide/3a0s-w68r-4qw1-0"
 
-pipe = StableDiffusionPipeline.from_ckpt(model_base, torch_dtype=torch.float16, use_safetensors=True)
+pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True)
 pipe.unet.load_attn_procs(lora_model_path) #working, commented to text base model------------------------------------
 #pipe.unet.load_attn_procs(lora_model_path, use_auth_token=True) test accessing a private model----------------------
 pipe.to("cuda")
@@ -43,7 +43,7 @@ css = """
 """
 
 def generate(prompt, neg_prompt):
-    images = pipe(prompt,negative_prompt=neg_prompt,num_inference_steps=25,guidance_scale=6).images
+    images = pipe(prompt,negative_prompt=neg_prompt,num_inference_steps=25,guidance_scale=6, cross_attention_kwargs={"scale": 0.5}).images
     return [(img, f"Image {i+1}") for i, img in enumerate(images)]