abhitopia
commited on
Commit
·
d355235
1
Parent(s):
5b49d65
bug fix
Browse files- code/inference.py +1 -0
code/inference.py
CHANGED
|
@@ -18,6 +18,7 @@ def predict_fn(input_data, model):
|
|
| 18 |
logger.info("input text: {}".format(input_data))
|
| 19 |
prediction = model(input_data)
|
| 20 |
logger.info("prediction: {}".format(input_data))
|
|
|
|
| 21 |
|
| 22 |
|
| 23 |
def input_fn(serialized_input_data, content_type=JSON_CONTENT_TYPE):
|
|
|
|
| 18 |
logger.info("input text: {}".format(input_data))
|
| 19 |
prediction = model(input_data)
|
| 20 |
logger.info("prediction: {}".format(input_data))
|
| 21 |
+
return prediction
|
| 22 |
|
| 23 |
|
| 24 |
def input_fn(serialized_input_data, content_type=JSON_CONTENT_TYPE):
|