Update app.py

app.py

@@ -23,14 +23,13 @@ def generate(
     temperature,
     top_p,
     top_k,
-    random_seed,
     seed,
 ):
 
     input_text = f"{system_prompt}, {prompt}"
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
 
-    if random_seed:
+    if seed == 0:
         seed = random.randint(1, 100000)
         torch.manual_seed(seed)
     else:
@@ -64,9 +63,7 @@ top_p = gr.Slider(value=1, minimum=0, maximum=2, step=0.05, interactive=True, la
 
 top_k = gr.Slider(value=1, minimum=1, maximum=100, step=1, interactive=True, label="Top K", info="Higher k means more diverse outputs by considering a range of tokens")
 
-use_random_seed = gr.Checkbox(value=False, label="Use Random Seed", info="Check to use a random seed which is a start point for the generation process")
-
-manual_seed = gr.Number(value=42, interactive=True, label="Manual Seed", info="A starting point to initiate the generation process", visible=lambda use_random_seed: not use_random_seed)
+manual_seed = gr.Number(value=42, interactive=True, label="Seed", info="A starting point to initiate the generation process, put 0 for a random one")
 
 
 examples = [
@@ -78,18 +75,16 @@ examples = [
         0.5,
         1,
         50,
-        False,
         42,
     ]
 ]
 
 gr.Interface(
     fn=generate,
-    inputs=[prompt, system_prompt, max_new_tokens, repetition_penalty, temperature, top_p, top_k, use_random_seed, manual_seed],
+    inputs=[prompt, system_prompt, max_new_tokens, repetition_penalty, temperature, top_p, top_k, seed],
     outputs=gr.Textbox(label="Better Prompt", interactive=True),
     title="SuperPrompt-v1",
     description="Make your prompts more detailed!<br>Model used: https://huggingface.co/roborovski/superprompt-v1<br>Hugging Face Space made by [Nick088](https://linktr.ee/Nick088)",
     examples=examples,
-    live=True,
     concurrency_limit=20,
 ).launch(show_api=False)