minor bug fixes

app.py

@@ -65,13 +65,15 @@ def infer(
     height,
     guidance_scale,
     num_inference_steps,
-    sliderspace_path,
+    slider_space,
+    discovered_directions,
     slider_scale,
     progress=gr.Progress(track_tqdm=True),
 ):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
+    sliderspace_path = f"sliderspace_weights/{slider_space}/{discovered_directions}",
     for net in networks:
         networks[net].load_state_dict(torch.load(sliderspace_path))
 
@@ -80,7 +82,7 @@ def infer(
 
     with networks[0]:
         pass
-
+    
     # original image
     generator = torch.Generator().manual_seed(seed)
     image = pipe(
@@ -142,7 +144,30 @@ with gr.Blocks(css=css) as demo:
         # New dropdowns side by side
         with gr.Row():
             slider_space = gr.Dropdown(
-                choices=["spaceship", "car", "person"],
+                choices= [
+                            "alien",
+                            "ancient ruins",
+                            "animal",
+                            "bike",
+                            "car",
+                            "Citadel",
+                            "coral",
+                            "cowboy",
+                            "face",
+                            "futuristic cities",
+                            "monster",
+                            "mystical creature",
+                            "planet",
+                            "plant",
+                            "robot",
+                            "sculpture",
+                            "sdxl_spaceship",
+                            "spaceship",
+                            "statue",
+                            "studio",
+                            "video game",
+                            "wizard"
+                        ],
                 label="SliderSpace",
                 value="spaceship"
             )
@@ -204,7 +229,7 @@ with gr.Blocks(css=css) as demo:
                 guidance_scale = gr.Slider(
                     label="Guidance scale",
                     minimum=0.0,
-                    maximum=10.0,
+                    maximum=2.0,
                     step=0.1,
                     value=0.0,  # Replace with defaults that work for your model
                 )
@@ -214,10 +239,10 @@ with gr.Blocks(css=css) as demo:
                     minimum=1,
                     maximum=50,
                     step=1,
-                    value=2,  # Replace with defaults that work for your model
+                    value=4,  # Replace with defaults that work for your model
                 )
 
-        gr.Examples(examples=examples, inputs=[prompt])
+        # gr.Examples(examples=exampleminor s, inputs=[prompt])
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,
@@ -230,7 +255,8 @@ with gr.Blocks(css=css) as demo:
             height,
             guidance_scale,
             num_inference_steps,
-            f"sliderspace_weights/{slider_space}/{discovered_directions}",
+            slider_space,
+            discovered_directions,
             slider_scale
         ],
         outputs=[result, slider_result, seed],