Spaces:

DeF0017
/

Style-Transfer-using-CycleGAN

Running

DeF0017 commited on Oct 12, 2024

Commit

463851a

verified ·

1 Parent(s): 22c9bf6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -75,10 +75,7 @@ NUM_WORKERS = 4
 NUM_EPOCHS = 0
 LOAD_MODEL = True
 SAVE_MODEL = False
-CHECKPOINT_GEN_B = "/kaggle/input/checkpoints/genB.pth.tar"
-CHECKPOINT_GEN_A = "/kaggle/input/checkpoints/genA.pth.tar"
-CHECKPOINT_DISC_A = "/kaggle/input/checkpoints/discA.pth.tar"
-CHECKPOINT_DISC_B = "/kaggle/input/checkpoints/discB.pth.tar"
 transforms = A.Compose(
     [

 NUM_EPOCHS = 0
 LOAD_MODEL = True
 SAVE_MODEL = False
+CHECKPOINT_GEN_A = f"{os.getcwd()}genA.pth.tar"
 transforms = A.Compose(
     [