Update app.py

app.py

@@ -13,10 +13,12 @@ tokenizer = AutoTokenizer.from_pretrained(decoder_checkpoint)
 caption_model = VisionEncoderDecoderModel.from_pretrained(model_checkpoint).to(device)
 
 # Load the Stable Diffusion model
-#diffusion_model = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
-diffusion_model = StableDiffusionPipeline.from_pretrained("jbilcke-hf/sdxl-cinematic-2")
+diffusion_model = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
 diffusion_model = diffusion_model.to(device)
 
+prompt = "a photo of an astronaut riding a horse on mars"
+image = pipe(prompt).images[0]  
+
 def predict(image):
     # Generate a caption from the image
     image = image.convert('RGB')