Update app.py

app.py

@@ -6,7 +6,7 @@ import math
 from pathlib import Path
 import sys
 sys.path.insert(1, './taming-transformers')
-from IPython import display
+#from IPython import display
 from base64 import b64encode
 from omegaconf import OmegaConf
 from PIL import Image
@@ -296,8 +296,8 @@ def inference(text, seed, step_size, max_iterations, width, height):
         losses_str = ', '.join(f'{loss.item():g}' for loss in losses)
         tqdm.write(f'i: {i}, loss: {sum(losses).item():g}, losses: {losses_str}')
         out = synth(z)
-        TF.to_pil_image(out[0].cpu()).save('progress.png')
-        display.display(display.Image('progress.png'))
+        #TF.to_pil_image(out[0].cpu()).save('progress.png')
+        #display.display(display.Image('progress.png'))
         res = nvidia_smi.nvmlDeviceGetUtilizationRates(handle)
         print(f'gpu: {res.gpu}%, gpu-mem: {res.memory}%')
     def ascend_txt():