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
|
|
|
|
|
|