Update app.py

app.py

@@ -10,33 +10,20 @@ import spaces
 extensions_dir = "./torch_extension/"
 os.environ["TORCH_EXTENSIONS_DIR"] = extensions_dir
 
-@spaces.GPU
-def test(prompt):
 
+@spaces.GPU
+def update(name):
     from networks.generator import Generator
-    
-    return prompt
-
-# gr.Interface(
-#     fn=test,
-#     inputs=gr.Text(),
-#     outputs=gr.Text(),
-# ).launch()
-
+    return f"Welcome to Gradio, {name}!"
 
-with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
-	# ... (input/output setup remains unchanged)
 
-	gr.HTML(load_file("assets/title.md"))
-	with gr.Row():
-		with gr.Accordion(open=False, label="Instruction"):
-			gr.Markdown(load_file("assets/instruction.md"))
-	
-	with gr.Interface(
-        fn=test,
-        inputs=gr.Text(),
-        outputs=gr.Text()
-    )
+with gr.Blocks() as demo:
+    gr.Markdown("Start typing below and then click **Run** to see the output.")
+    with gr.Row():
+        inp = gr.Textbox(placeholder="What is your name?")
+        out = gr.Textbox()
+    btn = gr.Button("Run")
+    btn.click(fn=update, inputs=inp, outputs=out)
 
 
 demo.launch(