Update app.py

app.py

@@ -232,6 +232,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)
+    print(device)
     print(padding.device)
     print(young.device)
     young_pad = torch.cat((young, padding), 1)