Update app.py

app.py

@@ -5,8 +5,8 @@ import random
 generator = pipeline('text-generation', model='jslin09/bloom-560m-finetuned-fraud')
 
 def generate(text):
-    seed = random.randint(10, 500)
-    set_seed(seed)
+    rnd_seed = random.randint(10, 500)
+    set_seed(rnd_seed)
     result = generator(text,
                        max_length=500,
                        num_return_sequences=1,
@@ -22,14 +22,19 @@ examples = [
     ["王大明意圖為自己不法所有，基於竊盜之犯意，"]
 ]
 
-demo = gr.Interface(
-    fn=generate,
-    title="Legal Document Drafting",
-    inputs=gr.components.Textbox(lines=5, label="輸入提示文字",placeholder=examples[0]),
-    outputs=gr.components.Textbox(label="生成的草稿"),
-    examples=examples,
-    cache_examples=False
-)
+with gr.Blocks() as demo:
+    gr.Markdown(
+    """
+    # <p style="text-align: center;">Legal Document Drafting</p>
+    """)
+    with gr.Row():
+        with gr.Column():
+            prompt = gr.components.Textbox(lines=5, label="輸入提示文字", placeholder=examples[0])
+            btn = gr.Button("生成草稿")
+            gr.Examples(examples, inputs=[prompt])
+        with gr.Column():    
+            result = gr.components.Textbox(label="生成的草稿")
+        btn.click(generate, inputs=[prompt], outputs=[result])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch() # 在遠端啟動時，需要 share=True 。