Update app.py

app.py

@@ -4,7 +4,7 @@ import random
 
 generator = pipeline('text-generation', model='jslin09/bloom-560m-finetuned-fraud')
 
-def generate(text):
+def rnd_generate(text):
     rnd_seed = random.randint(10, 500)
     set_seed(rnd_seed)
     result = generator(text,
@@ -16,6 +16,18 @@ def generate(text):
                        top_p=0.9)
     return result[0]["generated_text"]
 
+def generate(text):
+    set_seed(46)
+    result = generator(text,
+                       max_length=500,
+                       num_return_sequences=1,
+                       do_sample=True,
+                       temperature=0.75,
+                       top_k=50, 
+                       top_p=0.9)
+    return result[0]["generated_text"]
+
+
 examples = [
     ["闕很大明知金融帳戶之存摺、提款卡及密碼係供自己使用之重要理財工具，"],
     ["森上梅前明知其無資力支付酒店消費，亦無付款意願，竟意圖為自己不法之所有，"],
@@ -30,11 +42,13 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             prompt = gr.components.Textbox(lines=5, label="輸入提示文字", placeholder=examples[0])
-            btn = gr.Button("生成草稿")
+            btn = gr.Button("隨機生成草稿")
+            btn2 = gr.Button("固定生成草稿")
             gr.Examples(examples, inputs=[prompt])
         with gr.Column():    
             result = gr.components.Textbox(lines=15, label="生成的草稿")
-        btn.click(generate, inputs=[prompt], outputs=[result])
+        btn.click(rnd_generate, inputs=[prompt], outputs=[result])
+        btn2.click(generate, inputs=[prompt], outputs=[result])
 
 if __name__ == "__main__":
     demo.launch() # 在遠端啟動時，需要 share=True 。