Update app.py

app.py

@@ -1,5 +1,6 @@
 from huggingface_hub import model_info
-from diffusers import DiffusionPipeline
+from diffusers import StableDiffusionPipeline
+import torch
 
 # LoRA weights ~3 MB
 model_path = "cboard-org/testplaces"
@@ -9,7 +10,9 @@ model_base = info.cardData["base_model"]
 print("model is: "+model_base)   
 
 
-
-pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
-pipeline.load_lora_weights("cboard-org/testplaces")
+pipe = StableDiffusionPipeline.from_single_file(model_path,use_safetensors=True)
+pipe = pipe.to("cuda")
+prompt = "a photo of an astronaut riding a horse on mars"
+image = pipe(prompt).images[0]
+image.save("astronaut_rides_horse.png")