update

app.py

@@ -187,10 +187,10 @@ def generate(
     
     model.to(device)
     if device == "cuda":
-        return run(model, input_image, input_image_copy, width, height)
-    else:
         with torch.no_grad(), autocast("cuda"), model.ema_scope():
             return run(model, input_image, input_image_copy, width, height)
+    else:
+        return run(model, input_image, input_image_copy, width, height)
 
 
 def run(model, input_image, input_image_copy, width, height):