vasilee commited on
Commit
b16c32f
·
1 Parent(s): 96ab0e6

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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 {