Jaykintecblic commited on
Commit
4e28b18
1 Parent(s): 7e9a163

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -60,5 +60,5 @@ class EndpointHandler:
60
 
61
  generated_ids = self.model.generate(**inputs, bad_words_ids=self.bad_words_ids, max_length=4096)
62
  generated_text = self.processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
63
-
64
  return {"text": generated_text}
 
60
 
61
  generated_ids = self.model.generate(**inputs, bad_words_ids=self.bad_words_ids, max_length=4096)
62
  generated_text = self.processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
63
+ print(generated_text)
64
  return {"text": generated_text}