Ahsen Khaliq commited on
Commit
d67dc65
Β·
1 Parent(s): b61edf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ from pathlib import Path
7
  import sys
8
 
9
  sys.path.insert(1, './taming-transformers')
10
- from IPython import display
11
  from base64 import b64encode
12
  from omegaconf import OmegaConf
13
  from PIL import Image
@@ -358,7 +358,7 @@ def inference(text):
358
  tqdm.write(f'i: {i}, loss: {sum(losses).item():g}, losses: {losses_str}')
359
  out = synth(z)
360
  TF.to_pil_image(out[0].cpu()).save('progress.png')
361
- display.display(display.Image('progress.png'))
362
 
363
  def ascend_txt():
364
  # global i
 
7
  import sys
8
 
9
  sys.path.insert(1, './taming-transformers')
10
+ #from IPython import display
11
  from base64 import b64encode
12
  from omegaconf import OmegaConf
13
  from PIL import Image
 
358
  tqdm.write(f'i: {i}, loss: {sum(losses).item():g}, losses: {losses_str}')
359
  out = synth(z)
360
  TF.to_pil_image(out[0].cpu()).save('progress.png')
361
+ #display.display(display.Image('progress.png'))
362
 
363
  def ascend_txt():
364
  # global i