Update app.py

app.py

@@ -15,7 +15,7 @@ tokenizer = T5Tokenizer.from_pretrained("imranali291/flux-prompt-enhancer")
 model = T5ForConditionalGeneration.from_pretrained("imranali291/flux-prompt-enhancer", device_map="auto", torch_dtype="auto")
 model.to(device)
 
-def generate(your_prompt, task_prefix, max_new_tokens, repetition_penalty, temperature, model_precision_type, top_p, top_k, seed):
+def generate(your_prompt, max_new_tokens, repetition_penalty, temperature, model_precision_type, top_p, top_k, seed):
     
     if seed == 0:
         seed = random.randint(1, 2**32-1)
@@ -30,7 +30,7 @@ def generate(your_prompt, task_prefix, max_new_tokens, repetition_penalty, tempe
 
     repetition_penalty = float(repetition_penalty)
 
-    input_text = f"{task_prefix}: {your_prompt}"
+    input_text = f"{your_prompt}"
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
         
     outputs = model.generate(
@@ -69,7 +69,7 @@ examples = [
 
 gr.Interface(
     fn=generate,
-    inputs=[your_prompt, task_prefix, max_new_tokens, repetition_penalty, temperature, model_precision_type, top_p, top_k, seed],
+    inputs=[your_prompt, max_new_tokens, repetition_penalty, temperature, model_precision_type, top_p, top_k, seed],
     outputs=gr.Textbox(label="Prompt Enhancer"),
     title="Prompt Enhancer",
     description='Make your prompts more detailed!',