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; }