Nick088 commited on
Commit
d4c9f64
·
verified ·
1 Parent(s): e55a574

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1",
14
 
15
 
16
  def generate(
17
- prompt, history, temperature=temperature, max_new_tokens=max_new_tokens, repetition_penalty=repetition_penalty, temperature=temperature, top_p=top_p, top_k=top_k
18
  ):
19
 
20
  input_text = f"{prompt}, {history}"
 
14
 
15
 
16
  def generate(
17
+ prompt, history, max_new_tokens=max_new_tokens, repetition_penalty=repetition_penalty, temperature=temperature, top_p=top_p, top_k=top_k
18
  ):
19
 
20
  input_text = f"{prompt}, {history}"