Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
| 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(
|