convert conversion input to cuda

app.py

@@ -41,7 +41,7 @@ def infer(prompt, negative_prompt, image):
     # implement your inference function here
     inp = Image.fromarray(image)
 
-    cond_input = F.to_pil_image(conditioning_image_transforms(inp)[0])
+    cond_input = F.to_pil_image((conditioning_image_transforms(inp))[0].to("cuda"))
     #cond_input = T.ToPILImage(cond_input)
     
     output = pipe(