DeF0017 commited on
Commit
4d6ed90
·
verified ·
1 Parent(s): 24ebcd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -75,8 +75,8 @@ NUM_WORKERS = 4
75
  NUM_EPOCHS = 0
76
  LOAD_MODEL = True
77
  SAVE_MODEL = False
78
- CHECKPOINT_GEN_A = f"{os.getcwd()}genA.pth.tar"
79
- CHECKPOINT_GEN_B = f"{os.getcwd()}genB.pth.tar"
80
 
81
  transforms = A.Compose(
82
  [
 
75
  NUM_EPOCHS = 0
76
  LOAD_MODEL = True
77
  SAVE_MODEL = False
78
+ CHECKPOINT_GEN_A = f"{os.getcwd()}/genA.pth.tar"
79
+ CHECKPOINT_GEN_B = f"{os.getcwd()}/genB.pth.tar"
80
 
81
  transforms = A.Compose(
82
  [