Update app.py

app.py

@@ -211,10 +211,7 @@ def edit_inference(net, prompt, negative_prompt, guidance_scale, ddim_steps, see
     mean.to(device)
     std.to(device)
     v.to(device)
-    young.to(device)
-    pointy.to(device)
-    wavy.to(device)
-    thick.to(device)
+    
 
     
     weights = torch.load(net).to(device)
@@ -235,9 +232,9 @@ def edit_inference(net, prompt, negative_prompt, guidance_scale, ddim_steps, see
     pcs_edits = young.shape[1]
     padding =  torch.zeros((1,pcs_original-pcs_edits)).to(device)
     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)
+    pointy_pad = torch.cat((pointy.to(device), padding), 1)
+    wavy_pad = torch.cat((wavy.to(device), padding), 1)
+    thick_pad = torch.cat((thick.to(device), padding), 1)
             
         
     edited_weights = weights+a1*1e6*young_pad+a2*1e6*pointy_pad+a3*1e6*wavy_pad+a4*2e6*thick_pad