Jaykintecblic
commited on
Commit
•
4e28b18
1
Parent(s):
7e9a163
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}
|