Update app.py

app.py

@@ -41,9 +41,9 @@ def generate(prompt, option, progress=gr.Progress()):
     ckpt, step = opts[option]
     progress((0, step))
     
-    if pipe.device != device:
+    if pipe.device.type != device:
         pipe.to(device, dtype)
-    if safety_checker.device != device:
+    if safety_checker.device.type != device:
         safety_checker.to(device, dtype)
 
     if step != step_loaded: