Update app.py

app.py

@@ -1,18 +1,9 @@
-from huggingface_hub import model_info
-from diffusers import StableDiffusionPipeline
-import torch
+from diffusers import DiffusionPipeline
 
-# LoRA weights ~3 MB
-model_path = "cboard-org/testplaces"
+pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_safetensors=True)
 
-info = model_info(model_path)
-model_base = info.cardData["base_model"]
-print("model is: "+model_base)   
+pipeline.to("cuda")
 
-
-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")
+image = pipeline("An image of a squirrel in Picasso style").images[0]
+image