ArchitKohli commited on
Commit
b338992
·
1 Parent(s): 755b517

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -152,7 +152,7 @@ generator_f2m = Generator()
152
  discriminator_f = Discriminator()
153
  discriminator_m = Discriminator()
154
 
155
- checkpoint_dir = "https://huggingface.co/spaces/ArchitKohli/AnimizerGAN/tree/main"
156
  checkpoint_prefix = os.path.join(checkpoint_dir, "ckpt")
157
  checkpoint = tf.train.Checkpoint(generator_m2f=generator_m2f,
158
  generator_f2m=generator_f2m,
 
152
  discriminator_f = Discriminator()
153
  discriminator_m = Discriminator()
154
 
155
+ checkpoint_dir = "./"
156
  checkpoint_prefix = os.path.join(checkpoint_dir, "ckpt")
157
  checkpoint = tf.train.Checkpoint(generator_m2f=generator_m2f,
158
  generator_f2m=generator_f2m,