Added parameters

app.py

@@ -6,22 +6,23 @@ models = {
     "Metal" : pipeline("text-generation", "nschenone/metal-distil")
 }
 
-def generate(text, model):
-    max_length: int = 100
-    num_beams: int = 5
-    num_return_sequences: int = 1
-    no_repeat_ngram_size: int = 3
-    early_stopping: bool = True
-    skip_special_tokens: bool = True
-    temperature: float = 1.5
-
+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,
+):
     set_seed(0)
 
     generated = models[model](
         text_inputs=text,
         max_length=max_length,
         num_beams=num_beams,
-        num_return_sequences=num_return_sequences,
+        num_return_sequences=1,
         no_repeat_ngram_size=no_repeat_ngram_size,
         early_stopping=early_stopping,
         skip_special_tokens=skip_special_tokens,
@@ -39,7 +40,43 @@ iface = gr.Interface(
             choices=list(models.keys()),
             value=list(models.keys())[0],
             label="Model"
-        )
+        ),
+        gr.Slider(
+            minimum=50,
+            maximum=1000,
+            value=100,
+            step=10,
+            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,
+            value=1.5,
+            step=0.1,
+            label="Temperature"
+        ),
     ],
     outputs="text"
     )