Nick088 commited on
Commit
fe53ff3
·
verified ·
1 Parent(s): e1337fd

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, max_new_tokens=512, repetition_penalty=1.2, temperature=0.5, top_p=1, top_k=1
18
  ):
19
 
20
  input_text = f"{prompt}, {history}"
 
14
 
15
 
16
  def generate(
17
+ prompt, history, max_new_tokens=512, repetition_penalty=1.2, temperature=0.5, top_p=1, top_k=1,
18
  ):
19
 
20
  input_text = f"{prompt}, {history}"