Update handler.py

handler.py

@@ -1,5 +1,4 @@
 import torch
-import gradio as gr
 from PIL import Image
 import qrcode
 from pathlib import Path
@@ -128,7 +127,7 @@ class EndpointHandler():
                 controlnet_conditioning_scale=controlnet_conditioning_scale,
                 disable_progress_bar=True,
                 seed=generator,
-            ).cpu()
+            )
 
         result_image = (
             result_image.clamp(-1, 1).squeeze().permute(1, 2, 0).numpy() * 255