Update app.py

app.py

@@ -16,16 +16,8 @@ model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1",
 model.to(device)
     
 
-def generate(
-    prompt,
-    max_new_tokens,
-    repetition_penalty,
-    temperature,
-    top_p,
-    top_k,
-    seed
-):
-    input_text = f"Expand the following prompt to add more detail: {prompt}"
+def generate(your_prompt, max_new_tokens, repetition_penalty, temperature, top_p, top_k, seed):
+    input_text = f"Expand the following prompt to add more detail: {your_prompt}"
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
 
     if seed == 0:
@@ -49,7 +41,7 @@ def generate(
     return better_prompt
 
 
-prompt = gr.Textbox(label="Prompt", interactive=True)
+your_prompt = gr.Textbox(label="Your Prompt", interactive=True)
 
 max_new_tokens = gr.Slider(value=512, minimum=250, maximum=512, step=1, interactive=True, label="Max New Tokens", info="The maximum numbers of new tokens, controls how long is the output")
     
@@ -77,7 +69,7 @@ examples = [
 
 gr.Interface(
     fn=generate,
-    inputs=[prompt, max_new_tokens, repetition_penalty, temperature, top_p, top_k, seed],
+    inputs=[your_prompt, max_new_tokens, repetition_penalty, temperature, top_p, top_k, seed],
     outputs=gr.Textbox(label="Better Prompt"),
     title="SuperPrompt-v1",
     description='Make your prompts more detailed! <br> <a href="https://huggingface.co/roborovski/superprompt-v1">Model used</a> <br> <a href="https://brianfitzgerald.xyz/prompt-augmentation/">Model Blog</a> <br> Task Prefix: "Expand the following prompt to add more detail:" is already setted! <br> Hugging Face Space made by [Nick088](https://linktr.ee/Nick088)',