Update app.py

app.py

@@ -87,9 +87,9 @@ def run_all(prompt, steps, n_images, weight, clip_guided):
         if min(pred.shape[2:4]) < 256:
             pred = F.interpolate(pred, scale_factor=2, mode='bilinear', align_corners=False)
         clip_in = normalize(make_cutouts((pred + 1) / 2))
-        image_embeds = clip_model.encode_image(clip_in).view([args.cutn, x.shape[0], -1])
+        image_embeds = clip_model.encode_image(clip_in).view([16, x.shape[0], -1])
         losses = spherical_dist_loss(image_embeds, clip_embed[None])
-        loss = losses.mean(0).sum() * args.clip_guidance_scale
+        loss = losses.mean(0).sum() * 500.
         grad = -torch.autograd.grad(loss, x)[0]
         return grad