Update app.py

app.py

@@ -14,9 +14,23 @@ def predict_zh(sketch):
   return generated_text
   
  
-gr.Interface(
-    predict_zh,
-    inputs=gr.inputs.Textbox(lines=7, placeholder='Input your sketch', label='Input'),
-    outputs=gr.outputs.Label(num_top_classes=2),
-    title="Sketch-based Text Generation",
-).launch()
+with gr.Blocks() as demo:
+    gr.Markdown(
+        """
+        💡GENIUS – generating text using sketches! [Visit our github repo](https://github.com/beyondguo/genius)
+        The English version use `<mask>` as the mask token.
+        """)
+
+    with gr.Row():
+        with gr.Column():
+            model_input = gr.Textbox(lines=7, placeholder='Input your sketch', label='Input')
+            with gr.Row():
+                gen = gr.Button("Generate")
+                clr = gr.Button("Clear")
+               
+        outputs = gr.Textbox(lines=7, label='Output')
+        
+    gen.click(fn=predict_zh, inputs=[model_input], outputs=outputs)
+    clr.click(fn=lambda value: gr.update(value=""), inputs=clr, outputs=model_input)
+    
+demo.launch()