Vitrous commited on
Commit
d36192b
·
verified ·
1 Parent(s): ef40512

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ model, tokenizer = blender_model()
34
  def chat_response(msg_prompt: str) -> str:
35
  inputs = tokenizer([UTTERANCE], return_tensors="pt")
36
  reply_ids = model.generate(**inputs)
37
- outputs = tokenizer.batch_decode(reply_ids, skip_special_tokens=True)[0])
38
 
39
  return outputs
40
  except Exception as e:
 
34
  def chat_response(msg_prompt: str) -> str:
35
  inputs = tokenizer([UTTERANCE], return_tensors="pt")
36
  reply_ids = model.generate(**inputs)
37
+ outputs = tokenizer.batch_decode(reply_ids, skip_special_tokens=True)[0]
38
 
39
  return outputs
40
  except Exception as e: