Update app.py

app.py

@@ -105,7 +105,7 @@ w_stds = G.mapping(zs, None).std(0)
     
 def inference(text):
   target = clip_model.embed_text(text)
-  steps = 600
+  steps = 10
   seed = 2
   tf = Compose([
       Resize(224),
@@ -145,8 +145,6 @@ def inference(text):
     loop.set_postfix(loss=loss.item(), q_magnitude=q.std().item())
     q_ema = q_ema * 0.9 + q * 0.1
     image = G.synthesis(q_ema * w_stds + G.mapping.w_avg, noise_mode='const')
-    if i % 10 == 0:
-      display(TF.to_pil_image(tf(image)[0]))
     pil_image = TF.to_pil_image(image[0].add(1).div(2).clamp(0,1))
     #os.makedirs(f'samples/{timestring}', exist_ok=True)
     #pil_image.save(f'samples/{timestring}/{i:04}.jpg')