Spaces:

zanemotiwala
/

image-recognition-caption

Sleeping

zanemotiwala commited on Apr 14

Commit

b984c3e

•

1 Parent(s): 7d7443e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ caption_model = VisionEncoderDecoderModel.from_pretrained(model_checkpoint).to(d
 # Load the Stable Diffusion model
 #diffusion_model = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
-diffusion_model = StableDiffusionPipeline.from_pretrained("models/jbilcke-hf/sdxl-cinematic-2")
 diffusion_model = diffusion_model.to(device)
 def predict(image):

 # 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 = diffusion_model.to(device)
 def predict(image):