SentenceTransformer.h

Go to the documentation of this file.

#pragma once
#include "KrotosBinaryData.h"
struct bert_ctx;
 
namespace krotos
{
constexpr char modelFileName[] = ".ttpmodel";
class SentenceTransformer
{
 
  public:
    SentenceTransformer();
    ~SentenceTransformer();
    std::vector<float> encode(std::string sentence) const;
 
  private:
    File getModelPath() const;
    bool m_modelFileAvailable{false};
 
    void redirectPrint(File filePath, int& stdoutFd);
    void restorePrint(int stdoutFd);
 
    bert_ctx* bctx;
    int n_max_tokens = 0;
};
 
} // namespace krotos