Update app.py

app.py

@@ -27,7 +27,7 @@ def generate(your_prompt, max_new_tokens, repetition_penalty, temperature, model
 
     model.to(dtype)
 
-    input_text = f"Expand the following prompt to add more detail: {your_prompt}"
+    input_text = f"{task_prefix}: {your_prompt}"
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
         
     outputs = model.generate(
@@ -44,7 +44,9 @@ def generate(your_prompt, max_new_tokens, repetition_penalty, temperature, model
     return better_prompt
 
 
-your_prompt = gr.Textbox(label="Your Prompt", interactive=True)
+your_prompt = gr.Textbox(label="Your Prompt", info="Your Prompt that you wanna make better", interactive=True)
+
+task_prefix = gr.Textbox(label="Task Prefix", info="The prompt prefix for how the AI should make yours better",value="Expand the following prompt to add more detail", 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")