Update app.py

app.py

@@ -113,7 +113,7 @@ def text_to_face_generate(text):
     optimizer_pi = Adam((pi,), lr=5e-3)
 
     # optimization
-    for _ in range(10):
+    for _ in range(1):
         soft_pi = F.softmax(pi, dim=1)
         w_prime = soft_pi * w
         w_prime = w_prime.sum(dim=1)
@@ -160,7 +160,7 @@ def face_to_face_interpolate(image1, image2, interp_lambda=0.5):
     optimizer_w = SGLD((w,), lr=1e-1, momentum=0.99, noise_std=0.01, device=device)
 
     # optimization
-    for _ in range(10):
+    for _ in range(1):
         _, _, pred_clip_latent, _, _ = model(x=w, x_type='gan_latent', times_augment_pred_image=50)
         
         l = loss(target_clip_latent, pred_clip_latent)