Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -39,6 +39,6 @@ class EndpointHandler():
|
|
| 39 |
|
| 40 |
#logger.info(f"inputs: {inputs}")
|
| 41 |
|
| 42 |
-
outputs = self.model(inputs, **parameters)
|
| 43 |
logger.info(outputs)
|
| 44 |
return [{"generated_text": outputs["choices"][0]["text"]}]
|
|
|
|
| 39 |
|
| 40 |
#logger.info(f"inputs: {inputs}")
|
| 41 |
|
| 42 |
+
outputs = self.model.create_completion(inputs, **parameters)
|
| 43 |
logger.info(outputs)
|
| 44 |
return [{"generated_text": outputs["choices"][0]["text"]}]
|