Update app.py

app.py

@@ -68,39 +68,17 @@ def gpt_inference(base_url, model, openai_key, prompt):
     return img
 
 
+iface = gr.Interface(
+    fn=gpt_inference,
+    inputs=[gr.components.Textbox(), gr.components.Dropdown(choices=["gpt-3.5-turbo", "gpt-4"], label="Model"),
+            gr.components.Textbox(), gr.components.Textbox()],
+    outputs=gr.Image(),
+    title="SolidUI AI-generated visualization platform",
+    description="""
+    AI-generated visualization prototyping and editing platform, support 2D, 3D models, combined with LLM(Large Language Model) for quick editing.
+    GitHub: https://github.com/CloudOrc/SolidUI 
+    """,
+    labels=["Base URL", "Model", "OpenAI Key", "Prompt"]
 
-css = """
-body {
-    background-color: #F2F2F2;
-}
-"""
-
-title = """
-<h2>SolidUI AI-generated visualization platform</h2>
-<p>AI-generated visualization prototyping and editing platform Support 2D, 3D models, combined with LLM(Large Language Model) for quick editing.</p>
-<a target="_blank" href="https://github.com/CloudOrc/SolidUI">
-    <img src="https://img.shields.io/badge/-github-blue?logo=github" />
-</a>
-<a target="_blank" href="https://huggingface.co/spaces/CloudOrc/SolidUI">
-    <img src="https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face%20Spaces-blue" />
-</a>
-"""
-
-def layout_fn():
-    with gr.Blocks(css=css) as demo:
-        with gr.Column(elem_id="col-container"):
-            gr.HTML(title)
-            with gr.Form() as form:
-                form.inputs = [
-                    gr.components.Textbox(default="", label="Base URL"),
-                    gr.components.Dropdown(choices=["gpt-3.5-turbo", "gpt-4"], default="gpt-3.5-turbo", label="Model"),
-                    gr.components.Textbox(default="", label="OpenAI Key"),
-                    gr.components.Textbox(default="", label="Prompt")
-                ]
-                form.outputs = [gr.Image(label="Output Image")]
-                form.submit_button = gr.Button('Run')
-    return demo
-
-
-iface = gr.Interface(fn=gpt_inference, inputs=[], outputs=[], layout=layout_fn)
+)
 iface.launch()