public ActionResult <IEnumerable <string> > PredictPost(QuestionViewModel question)
{
var(tokens, probability) = _bertModel.Predict(question.Context, question.Question);
return(Ok(new PredictionViewModel()
{
Tokens = tokens,
Probability = probability
}));
}
static void Main(string[] args)
{
var modelConfig = new BertModelConfiguration()
{
VocabularyFile = "Model/vocab.txt",
ModelPath = "Model/bertsquad-10.onnx"
};
var model = new BertModel(modelConfig);
model.Initialize();
var(tokens, probability) = model.Predict(args[0], args[1]);
System.Console.WriteLine(JsonSerializer.Serialize(new
{
Probability = probability,
Tokens = tokens
}));
}