Update app.py

app.py

@@ -234,7 +234,7 @@ def edit_inference(net, prompt, negative_prompt, guidance_scale, ddim_steps, see
     pcs_original = weights.shape[1]
     pcs_edits = young.shape[1]
     padding =  torch.zeros((1,pcs_original-pcs_edits)).to(device)
-    young_pad = torch.cat((young, padding), 1)
+    young_pad = torch.cat((young.to(device), padding), 1)
     pointy_pad = torch.cat((pointy, padding), 1)
     wavy_pad = torch.cat((wavy, padding), 1)
     thick_pad = torch.cat((thick, padding), 1)