Spaces:
Paused
Paused
Commit
·
0dc748d
1
Parent(s):
1df84d8
Update answerer.py
Browse files- answerer.py +3 -3
answerer.py
CHANGED
|
@@ -20,6 +20,7 @@ class Answerer:
|
|
| 20 |
repetitiveness = .3,
|
| 21 |
diversity = 0,
|
| 22 |
_count_penalty = 1,
|
|
|
|
| 23 |
):
|
| 24 |
args = PIPELINE_ARGS(
|
| 25 |
temperature=chaos,
|
|
@@ -67,8 +68,7 @@ class Answerer:
|
|
| 67 |
if "\ufffd" not in tmp:
|
| 68 |
tokens.clear()
|
| 69 |
result += tmp
|
| 70 |
-
|
| 71 |
-
yield result
|
| 72 |
|
| 73 |
tokens.clear()
|
| 74 |
occurrences.clear()
|
|
@@ -76,5 +76,5 @@ class Answerer:
|
|
| 76 |
del occurrences, tokens, current_token, state
|
| 77 |
gc.collect()
|
| 78 |
|
| 79 |
-
|
| 80 |
|
|
|
|
| 20 |
repetitiveness = .3,
|
| 21 |
diversity = 0,
|
| 22 |
_count_penalty = 1,
|
| 23 |
+
callback = None,
|
| 24 |
):
|
| 25 |
args = PIPELINE_ARGS(
|
| 26 |
temperature=chaos,
|
|
|
|
| 68 |
if "\ufffd" not in tmp:
|
| 69 |
tokens.clear()
|
| 70 |
result += tmp
|
| 71 |
+
if(callback) callback(result)
|
|
|
|
| 72 |
|
| 73 |
tokens.clear()
|
| 74 |
occurrences.clear()
|
|
|
|
| 76 |
del occurrences, tokens, current_token, state
|
| 77 |
gc.collect()
|
| 78 |
|
| 79 |
+
return result
|
| 80 |
|