Add application file

app.py

@@ -8,7 +8,8 @@ import numpy as np
 model_id = "CompVis/stable-diffusion-v1-4"
 default_scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
 pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=default_scheduler, torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = pipe.to(device)
 
 # Scheduler options
 schedulers = {