RohitGandikota commited on
Commit
989fccf
·
verified ·
1 Parent(s): 932f63b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ def infer(
73
  if randomize_seed:
74
  seed = random.randint(0, MAX_SEED)
75
 
76
- sliderspace_path = f"sliderspace_weights/{slider_space}/slider_{int(discovered_directions[-1])-1}.pt"
77
 
78
  for net in networks:
79
  networks[net].load_state_dict(torch.load(sliderspace_path))
 
73
  if randomize_seed:
74
  seed = random.randint(0, MAX_SEED)
75
 
76
+ sliderspace_path = f"sliderspace_weights/{slider_space}/slider_{int(discovered_directions.split(' ')[-1])-1}.pt"
77
 
78
  for net in networks:
79
  networks[net].load_state_dict(torch.load(sliderspace_path))