Update app.py

app.py

@@ -21,7 +21,6 @@ with gr.Blocks() as demo:  # Removed the css argument
         """<h1><img src="https://i.imgur.com/vT48NAO.png" alt="LoRA"> LoRA the Explorer</h1>""",
         elem_id="title",
     )
-    selected_state = gr.State()
     gallery = gr.Gallery(
         value=[(item["image"], item["title"]) for item in loras],
         label="LoRA Gallery",
@@ -36,8 +35,8 @@ with gr.Blocks() as demo:  # Removed the css argument
     result = gr.Image(interactive=False, label="Generated Image", elem_id="result-image")
 
     # Define the function to run when the button is clicked
-    def run_lora(prompt, weight, selected_state):
-        selected_lora = loras[selected_state]
+    def run_lora(prompt, weight):
+        selected_lora = loras[0]  # You may need to adjust this index if you have multiple models
         api_url = f"https://api-inference.huggingface.co/models/{selected_lora['repo']}"
         trigger_word = selected_lora["trigger_word"]
         token = os.getenv("API_TOKEN")  # This will read the API token set in your managed environment
@@ -51,6 +50,6 @@ with gr.Blocks() as demo:  # Removed the css argument
     print("Starting Gradio UI...")
     gr.Interface(
         fn=run_lora,
-        inputs=[prompt, weight, selected_state],
+        inputs=[prompt, weight],
         outputs=[result],
     ).launch()