harpreetsahota commited on
Commit
0d4a6a6
1 Parent(s): 9a81d74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def generate_model_response(message: str) -> str:
61
  temperature=0.1,
62
  streamer=streamer
63
  )
64
- # return tokenizer.decode(output[0], skip_special_tokens=True)
65
 
66
  def extract_response_content(full_response: str) -> str:
67
  response_start_index = full_response.find("### Assistant:")
 
61
  temperature=0.1,
62
  streamer=streamer
63
  )
64
+ return tokenizer.decode(output[0], skip_special_tokens=True)
65
 
66
  def extract_response_content(full_response: str) -> str:
67
  response_start_index = full_response.find("### Assistant:")