Spaces:
Runtime error
Runtime error
Update rpc.py
Browse files
rpc.py
CHANGED
@@ -199,8 +199,6 @@ def generate(text, use_rpc=True, max_tokens=128):
|
|
199 |
tok = tf.argmax(res, axis=-1).numpy().tolist()
|
200 |
|
201 |
enc_text += [tok]
|
202 |
-
|
203 |
-
|
204 |
-
|
205 |
-
|
206 |
-
yield tok_str
|
|
|
199 |
tok = tf.argmax(res, axis=-1).numpy().tolist()
|
200 |
|
201 |
enc_text += [tok]
|
202 |
+
response = tokenizer.decode(enc_text)
|
203 |
+
|
204 |
+
yield response
|
|
|
|