Added seed and removed parameters that will not change from UI

app.py

@@ -2,27 +2,36 @@ import gradio as gr
 from transformers import pipeline, set_seed
 
 models = {
-    "Rap" : pipeline("text-generation", "nschenone/rap-distil"),
-    "Metal" : pipeline("text-generation", "nschenone/metal-distil")
+    "Rap" : pipeline(
+        task="text-generation",
+        model="nschenone/rap-distil"
+    ),
+    "Metal" : pipeline(
+        task="text-generation",
+        model="nschenone/metal-distil"
+    )
 }
 
 def generate(
     text: str,
     model: str,
     max_length: int = 100,
-    num_beams: int = 5,
-    no_repeat_ngram_size: int = 3,
-    early_stopping: bool = True,
-    skip_special_tokens: bool = True,
     temperature: float = 1.5,
+    seed: int = 0
 ):
-    set_seed(0)
+    num_beams: int = 5
+    num_return_sequences: int = 1
+    no_repeat_ngram_size: int = 3
+    early_stopping: bool = True
+    skip_special_tokens: bool = True
+
+    set_seed(seed)
 
     generated = models[model](
         text_inputs=text,
         max_length=max_length,
+        num_return_sequences=num_return_sequences,
         num_beams=num_beams,
-        num_return_sequences=1,
         no_repeat_ngram_size=no_repeat_ngram_size,
         early_stopping=early_stopping,
         skip_special_tokens=skip_special_tokens,
@@ -35,7 +44,11 @@ def generate(
 iface = gr.Interface(
     fn=generate,
     inputs=[
-        "text",
+        gr.Textbox(
+            value="[Verse]",
+            placeholder="Input text...",
+            label="Input Text"
+        ),
         gr.Dropdown(
             choices=list(models.keys()),
             value=list(models.keys())[0],
@@ -49,34 +62,17 @@ iface = gr.Interface(
             label="Max Length"
         ),
         gr.Slider(
-            minimum=1,
-            maximum=5,
-            value=5,
-            step=1,
-            label="Num Beams"
-        ),
-        gr.Slider(
-            minimum=1,
-            maximum=3,
-            value=3,
-            step=1,
-            label="No Repeat N-Gram Size"
-        ),
-        gr.Checkbox(
-            value=True,
-            label="Early Stopping"
-        ),
-        gr.Checkbox(
-            value=True,
-            label="Skip Special Tokens"
-        ),
-        gr.Slider(
-            minimum=0,
-            maximum=2,
+            minimum=0.4,
+            maximum=1.9,
             value=1.5,
             step=0.1,
             label="Temperature"
         ),
+        gr.Number(
+            value=0,
+            precision=0,
+            label="Seed"
+        ),
     ],
     outputs="text"
     )