Update handler.py
Browse files- 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}
|