Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -83,7 +83,7 @@ async def inference(request: InferenceRequest):
|
|
| 83 |
|
| 84 |
output_tokens = results[0].hypotheses[0]
|
| 85 |
output_text = inferenceTokenizer.decode(
|
| 86 |
-
inferenceTokenizer.convert_tokens_to_ids(output_tokens))
|
| 87 |
|
| 88 |
# create response
|
| 89 |
return {
|
|
|
|
| 83 |
|
| 84 |
output_tokens = results[0].hypotheses[0]
|
| 85 |
output_text = inferenceTokenizer.decode(
|
| 86 |
+
inferenceTokenizer.convert_tokens_to_ids(output_tokens), skip_special_tokens=True)
|
| 87 |
|
| 88 |
# create response
|
| 89 |
return {
|