Spaces:
Running
Running
Commit
·
25bb492
1
Parent(s):
96b5d9c
work on low CPU
Browse files- .gitattributes +1 -1
- .gitignore +1 -0
- 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=
|
32 |
-
num_beam_groups=
|
33 |
-
num_return_sequences=
|
34 |
-
repetition_penalty=
|
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 |
)
|