Update app.py

app.py

@@ -30,30 +30,35 @@ def run_lora(prompt, weight):
     return Image.open(image_bytes)
 
 # Gradio UI
-print("Before Gradio Blocks")
-with gr.Blocks(css="custom.css") as demo:
-    print("Inside Gradio Blocks")
-    title = gr.HTML(
-        """<h1><img src="https://i.imgur.com/vT48NAO.png" alt="LoRA"> LoRA the Explorer</h1>""",
-        elem_id="title",
-    )
-    gallery = gr.Gallery(
-        value=[(item["image"], item["title"]) for item in loras],
-        label="LoRA Gallery",
-        allow_preview=False,
-        columns=3,
-        elem_id="gallery",
-        show_share_button=False
-    )
-    prompt = gr.Textbox(label="Prompt", show_label=False, lines=1, max_lines=1, placeholder="Type a prompt after selecting a LoRA", elem_id="prompt")
-    advanced_options = gr.Accordion("Advanced options", open=False)
-    weight = gr.Slider(0, 10, value=1, step=0.1, label="LoRA weight")
-    result = gr.Image(interactive=False, label="Generated Image", elem_id="result-image")
-
 print("Before Gradio Interface")
+
+title = gr.HTML(
+    """<h1>LoRA the Explorer</h1>""",
+    elem_id="title",
+)
+
+gallery = gr.Gallery(
+    value=[(item["image"], item["title"]) for item in loras],
+    label="LoRA Gallery",
+    allow_preview=False,
+    columns=3,
+    elem_id="gallery",
+    show_share_button=False
+)
+
+prompt = gr.Textbox(label="Prompt", show_label=False, lines=1, max_lines=1, placeholder="Type a prompt after selecting a LoRA", elem_id="prompt")
+
+advanced_options = gr.Accordion("Advanced options", open=False)
+
+weight = gr.Slider(0, 10, value=1, step=0.1, label="LoRA weight")
+
+result = gr.Image(interactive=False, label="Generated Image", elem_id="result-image")
+
 gr.Interface(
     fn=run_lora,
-    inputs=[prompt, weight],
+    inputs=[title, gallery, prompt, advanced_options, weight],
     outputs=[result],
+    css="custom.css"  # Make sure your custom CSS file is in the same directory
 ).launch()
+
 print("After Gradio Interface")