Update app.py

app.py

@@ -14,12 +14,12 @@ model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1",
 
 
 def generate(
-    prompt, history, temperature=0.9, max_new_tokens=250, repetition_penalty=1.0,
+    prompt, history, temperature=temperature, max_new_tokens=max_new_tokens, repetition_penalty=repetition_penalty, temperature=temperature, top_p=top_p, top_k=top_k
 ):
     
     input_text = f"{prompt}, {history}"
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
-    outputs = model.generate(input_ids, max_new_tokens=max_new_tokens, repetition_penalty=repetition_penalty)
+    outputs = model.generate(input_ids, max_new_tokens=max_new_tokens, repetition_penalty=repetition_penalty, temperature=temperature, top_p=top_p, top_k=top_k)
     better_prompt = tokenizer.decode(outputs[0])
     return better_prompt