seeded

app.py

@@ -1,6 +1,6 @@
 import gc 
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, set_seed
 
 pipe = pipeline('text-generation', framework='pt', model='akhooli/ap2023', tokenizer='akhooli/ap2023')
 #gc.collect()
@@ -29,7 +29,8 @@ notes = """
 - Feedback (Twitter): @akhooli (https://twitter.com/akhooli/status/1611025232201977859) 
 - Note/Disclaimer: may generate unaccepted or inappropriate content. Use at your own risk.
 """
-def sayPoetry(prompt, temp=1.0, topk = 50, topp = 1.0, penalty=1.0):
+def sayPoetry(prompt, temp=1.0, topk = 50, topp = 1.0, penalty=1.0, seed=114):
+    set_seed(seed)
     gen = pipe(prompt, max_length=96, do_sample=True, temperature=temp, top_k=topk, top_p=topp, repetition_penalty=penalty,
               min_length = 64, no_repeat_ngram_size = 3, return_full_text=True,
                num_beams=5, num_return_sequences=1)[0]["generated_text"]
@@ -44,6 +45,7 @@ poetry = gr.Interface(fn=sayPoetry,
                         gr.Slider(25, 100, step=1,value=50, label='control  top k'),
                         gr.Slider(0.80, 1.0, step=0.01,value=1.0, label='control  top p'),
                         gr.Slider(0.90, 1.20, step=0.01,value=1.0, label='control  penalty'),
+                        gr.Number(default=139750, label='Seed'),
                     ], 
                 outputs=[gr.Textbox(label="Generated Poetry:")],