Update app.py

app.py

@@ -1,5 +1,5 @@
-#TODO: set seed and argument for it
-#TODO: use transformer library directly
+#TODO: verify seed works
+#TODO: integrate into markdown
 import gradio as gr
 from gradio import mix
 from transformers import pipeline, set_seed
@@ -18,7 +18,7 @@ io2 = gr.Interface.load("huggingface/gpt2-large")
 
 #io4 = gr.Interface.load("huggingface/gpt2-xl")
 
-def inference(text, random_seed):
+def inference(text, seed):
     """
     if model == "gpt2-large":
         outtext = io2(text)
@@ -38,7 +38,7 @@ def inference(text, random_seed):
 
 gr.Interface(
     inference, 
-    [gr.inputs.Textbox(label="input", placeholder="trustworthy artificial intelligence"), gr.inputs.Number(label="random_seed")],
+    [gr.inputs.Textbox(label="input", placeholder="trustworthy artificial intelligence"), gr.inputs.Number(label="seed")],
     #,gr.inputs.Dropdown(choices=["distilgpt2","gpt2-medium","gpt2-large","gpt2-xl"], type="value", default="gpt2-medium", label="model")], 
     gr.outputs.Textbox(label="gpt-2 proposal"),
     #title=title,