abhitopia
/

question-answer-generation

Text2Text Generation

question-answer-generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

abhitopia commited on Aug 31, 2022

Commit

d355235

•

1 Parent(s): 5b49d65

bug fix

Files changed (1) hide show

code/inference.py +1 -0

code/inference.py CHANGED Viewed

@@ -18,6 +18,7 @@ def predict_fn(input_data, model):
     logger.info("input text: {}".format(input_data))
     prediction = model(input_data)
     logger.info("prediction: {}".format(input_data))
 def input_fn(serialized_input_data, content_type=JSON_CONTENT_TYPE):

     logger.info("input text: {}".format(input_data))
     prediction = model(input_data)
     logger.info("prediction: {}".format(input_data))
+    return prediction
 def input_fn(serialized_input_data, content_type=JSON_CONTENT_TYPE):