ravindravala commited on
Commit
25bb492
·
1 Parent(s): 96b5d9c

work on low CPU

Browse files
Files changed (3) hide show
  1. .gitattributes +1 -1
  2. .gitignore +1 -0
  3. app.py +4 -4
.gitattributes CHANGED
@@ -32,4 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
32
  *.xz filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
- *tfevents* filter=lfs diff=lfs merge=lfs -text
 
32
  *.xz filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ myenv
app.py CHANGED
@@ -28,10 +28,10 @@ def humanize_text(text, temperature=0.7, max_length=512):
28
  input_ids,
29
  max_length=max_length,
30
  temperature=temperature,
31
- num_beams=3,
32
- num_beam_groups=3,
33
- num_return_sequences=3,
34
- repetition_penalty=3.0,
35
  diversity_penalty=0.5,
36
  no_repeat_ngram_size=2,
37
  )
 
28
  input_ids,
29
  max_length=max_length,
30
  temperature=temperature,
31
+ num_beams=1,
32
+ num_beam_groups=1,
33
+ num_return_sequences=1,
34
+ repetition_penalty=2.0,
35
  diversity_penalty=0.5,
36
  no_repeat_ngram_size=2,
37
  )