RohitGandikota commited on
Commit
0c8d966
·
verified ·
1 Parent(s): ad0458b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -73,7 +73,8 @@ 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
  for net in networks:
78
  networks[net].load_state_dict(torch.load(sliderspace_path))
79
 
@@ -127,7 +128,7 @@ css = """
127
 
128
  with gr.Blocks(css=css) as demo:
129
  with gr.Column(elem_id="col-container"):
130
- gr.Markdown(" # SliderSpace: Automatic Slider Discovery from Diffusion Model's Knowledge")
131
 
132
  with gr.Row():
133
  prompt = gr.Text(
 
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))
80
 
 
128
 
129
  with gr.Blocks(css=css) as demo:
130
  with gr.Column(elem_id="col-container"):
131
+ gr.Markdown(" # SliderSpace Demo")
132
 
133
  with gr.Row():
134
  prompt = gr.Text(