fix

app.py

@@ -190,7 +190,7 @@ def combine_components_slice(model, gd, im1, im2, indices=None, sample_method='d
 
 def combine_images(im1, im2):
     sample_method = 'ddim'
-    result = combine_components_slice(celeb_model, GD[sample_method], im1, im2, indices='1,0,1,0', sample_method=sample_method, num_images=1)
+    result = combine_components_slice(celeb_model, GD[sample_method], im1, im2, indices='1,0,1,0', sample_method=sample_method, num_images=1, device=device)
     return result.permute(1, 2, 0).numpy()