Update app.py

app.py

@@ -16,7 +16,7 @@ def query(payload, api_url, token):
     return io.BytesIO(response.content)
 
 # Gradio UI
-with gr.Blocks(css="custom.css") as demo:
+with gr.Blocks() as demo:  # Removed the css argument
     title = gr.HTML(
         """<h1><img src="https://i.imgur.com/vT48NAO.png" alt="LoRA"> LoRA the Explorer</h1>""",
         elem_id="title",
@@ -49,8 +49,8 @@ with gr.Blocks(css="custom.css") as demo:
         return Image.open(image_bytes)
 
     print("Starting Gradio UI...")
-    prompt.submit(
+    gr.Interface(
         fn=run_lora,
         inputs=[prompt, weight, selected_state],
         outputs=[result],
-    )
+    ).launch()