Hugging Face's logo

Spaces:
nmud19
/
Sketch2ColourDemo
Running

App Files Community
2
Nikhil Mudhalwadkar commited on
Commit
b4a3300
·
1 Parent(s): 8866157

use new model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -148,7 +148,7 @@ class Generator(nn.Module):
148
 
149
 
150
  """ Load the model """
151
- model_checkpoint_path = "model/lightning_bolts_model/epoch=8-step=8010.ckpt"
152
  # model_checkpoint_path = "model/pix2pix_lightning_model/version_0/checkpoints/epoch=199-step=355600.ckpt"
153
  # model_checkpoint_path = "model/pix2pix_lightning_model/gen.pth"
154
 
 
148
 
149
 
150
  """ Load the model """
151
+ model_checkpoint_path = "model/lightning_bolts_model/epoch=99-step=89000.ckpt"
152
  # model_checkpoint_path = "model/pix2pix_lightning_model/version_0/checkpoints/epoch=199-step=355600.ckpt"
153
  # model_checkpoint_path = "model/pix2pix_lightning_model/gen.pth"
154