Update app.py

app.py

@@ -21,6 +21,7 @@ def generate(
     temperature,
     top_p,
     top_k,
+    seed_checkbox,
     seed,
     model_path="roborovski/superprompt-v1",
     dtype="fp16",
@@ -40,7 +41,9 @@ def generate(
     input_text = f"{prompt}, {history}"
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
 
-    torch.manual_seed(seed)
+    if seed_checkbox:
+        torch.manual_seed(seed)
+
     outputs = model.generate(
         input_ids,
         max_new_tokens=max_new_tokens,
@@ -100,6 +103,11 @@ additional_inputs = [
         label="Top K",
         info="Higher k means more diverse outputs by considering a range of tokens",
     ),
+    gr.Checkbox(
+        value=False,
+        label="Use Random Seed",
+        info="Check to use a random seed for the generation process",
+    ),
     gr.Number(
         value=42,
         interactive=True,
@@ -123,6 +131,7 @@ examples = [
         None,
         None,
         None,
+        False,
         None,
         "roborovski/superprompt-v1",
         "fp16",