augment paths

1 files changed, 2 insertions, 2 deletions

diff --git a/live-mogrify.py b/live-mogrify.py
index 5cd7f67..9fba009 100644
--- a/live-mogrify.py
+++ b/live-mogrify.py
@@ -133,13 +133,13 @@ def process_image(opt, data_opt, im):
 
 def list_checkpoints(payload):
   print("> list checkpoints")
-  return sorted([f.split('/')[2] for f in glob.glob('./checkpoints/' + payload + '/*/latest_net_G.pth')])
+  return sorted([f.split('/')[3] for f in glob.glob('./checkpoints/' + payload + '/*/latest_net_G.pth')])
 
 def list_epochs(path):
   print("> list epochs for {}".format(path))
   if not os.path.exists(os.path.join('./checkpoints/', path)):
     return "not found"
-  return sorted([f.split('/')[3].replace('_net_G.pth','') for f in glob.glob('./checkpoints/' + path + '/*_net_G.pth')])
+  return sorted([f.split('/')[4].replace('_net_G.pth','') for f in glob.glob('./checkpoints/' + path + '/*_net_G.pth')])
 
 def list_sequences(module):
   print("> list sequences")