Norod78 commited on
Commit
9d56fd7
1 Parent(s): bad33a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def text_generation(input_text = ''):
35
  top_p=top_p,
36
  temperature=temperature,
37
  do_sample=True,
38
- repetition_penalty=2.0,
39
  num_return_sequences=1)
40
  parsed_text = generated_text[0]["generated_text"].replace("<|startoftext|>", "").replace("\r","").replace("\n\n", "\n").replace("\t", " ").replace("<|pad|>", " * ").replace("\"\"", "\"").strip()
41
  print("parsed_text = \"" + parsed_text + "\" (seed = " + str(global_seed) + ")")
 
35
  top_p=top_p,
36
  temperature=temperature,
37
  do_sample=True,
38
+ repetition_penalty=2.2,
39
  num_return_sequences=1)
40
  parsed_text = generated_text[0]["generated_text"].replace("<|startoftext|>", "").replace("\r","").replace("\n\n", "\n").replace("\t", " ").replace("<|pad|>", " * ").replace("\"\"", "\"").strip()
41
  print("parsed_text = \"" + parsed_text + "\" (seed = " + str(global_seed) + ")")