Update answerer.py

answerer.py

@@ -1,4 +1,4 @@
-from typing import Callable, Dict, List
+from typing import Dict, List
 import os, gc
 from rwkv.model import RWKV
 from rwkv.utils import PIPELINE, PIPELINE_ARGS
@@ -20,7 +20,6 @@ class Answerer:
     repetitiveness = .3,
     diversity = 0,
     _count_penalty = 1,
-    callback = None,
   ):
     args = PIPELINE_ARGS(
       temperature=chaos,
@@ -68,7 +67,7 @@ class Answerer:
       if "\ufffd" not in tmp:
         tokens.clear()
         result += tmp
-        if(callback): callback(result)
+        yield result
 
     tokens.clear()
     occurrences.clear()
@@ -76,5 +75,5 @@ class Answerer:
     del occurrences, tokens, current_token, state 
     gc.collect()
 
-    return result
+    yield result