Update app_sd.py

app_sd.py

@@ -10,7 +10,7 @@ import os
 MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
 YOUR_TOKEN=MY_SECRET_TOKEN
 # load the pipeline
-device = "cuda"
+device = "cpu"
 model_id_or_path = "CompVis/stable-diffusion-v1-4"
 pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
     model_id_or_path,
@@ -31,7 +31,7 @@ init_image = init_image.resize((768, 512))
 
 prompt = "Lively, illustration of a [[[<king::4>]]], portrait, fantasy, intricate, Scenic, hyperdetailed, hyper realistic <king-hearthstone>, unreal engine, 4k, smooth, sharp focus, intricate, cinematic lighting, highly detailed, octane, digital painting, artstation, concept art, vibrant colors, Cinema4D, WLOP, 3d render, in the style of hearthstone::5 art by Artgerm and greg rutkowski and magali villeneuve, martina jackova, Giger"
 
-with autocast("cuda"):
+with autocast(device): #"cuda"):
     images = pipe(prompt=prompt, init_image=init_image, strength=0.75, guidance_scale=7.5).images
 
 images[0].save("fantasy_landscape.png")