typo

artist_lib.py

@@ -87,6 +87,7 @@ def generate_tone(note, octave, duration):
     return sr, audio
 
 def draw(inp, this_model, force_new):
+    device = "cuda" if torch.cuda.is_available() else "cpu"
     drawing = inp
     if this_model == "stable-diffusion-2":
         this_model_addr = "stabilityai/stable-diffusion-2"
@@ -111,7 +112,7 @@ def draw(inp, this_model, force_new):
     print("generating drawing '", drawing, "'", drawing_filename)
     pipe = StableDiffusionPipeline.from_pretrained(this_model_addr, torch_dtype=torch.float16)
     pipe.enable_attention_slicing()
-    pipe = pipe.to("cuda")
+    pipe = pipe.to(device)
     image = pipe(drawing).images[0]  
     image.seek(0)
     image.save(drawing_filename)