Amitontheweb commited on
Commit
e1e6e43
·
1 Parent(s): c3efef2

Removed temperature from def paraphrase and outputs field.

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -20,7 +20,6 @@ def paraphrase(
20
  repetition_penalty=10.0,
21
  diversity_penalty=3.0,
22
  no_repeat_ngram_size=2,
23
- temperature=0.7,
24
  max_length=5000
25
  ):
26
  input_ids = tokenizer(
@@ -31,7 +30,7 @@ def paraphrase(
31
  ).input_ids
32
 
33
  outputs = model.generate(
34
- input_ids, temperature=temperature, repetition_penalty=repetition_penalty,
35
  num_return_sequences=num_return_sequences, no_repeat_ngram_size=no_repeat_ngram_size,
36
  num_beams=num_beams, num_beam_groups=num_beam_groups,
37
  max_length=max_length, diversity_penalty=diversity_penalty
 
20
  repetition_penalty=10.0,
21
  diversity_penalty=3.0,
22
  no_repeat_ngram_size=2,
 
23
  max_length=5000
24
  ):
25
  input_ids = tokenizer(
 
30
  ).input_ids
31
 
32
  outputs = model.generate(
33
+ input_ids, repetition_penalty=repetition_penalty,
34
  num_return_sequences=num_return_sequences, no_repeat_ngram_size=no_repeat_ngram_size,
35
  num_beams=num_beams, num_beam_groups=num_beam_groups,
36
  max_length=max_length, diversity_penalty=diversity_penalty