public void ConfigureServices(IServiceCollection services)
{
services.AddSingleton(o =>
{
var modelConfig = new BertModelConfiguration()
{
VocabularyFile = "Model/vocab.txt",
ModelPath = "Model/bertsquad-10.onnx"
};
var model = new BertModel(modelConfig);
model.Initialize();
return(model);
});
services.AddControllers();
}
public void ConfigureServices(IServiceCollection services)
{
services.AddMvc().SetCompatibilityVersion(CompatibilityVersion.Version_2_2);
services.AddSingleton(o =>
{
var modelConfig = new BertModelConfiguration()
{
VocabularyFile = "Model/vocab.txt",
ModelPath = "Model/bertsquad8.onnx"
};
var model = new BertModel(modelConfig);
model.Initialize();
return(model);
});
}
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
}));
}
public PredictionController(BertModel bertModel)
{
_bertModel = bertModel;
}
