Add std::istream variant to Load and LoadOrDie

Added for SentencePieceProcessor:
virtual bool Load(std::istream &istream);
virtual void LoadOrDie(std::istream &istream);

1 files changed, 8 insertions, 0 deletions

diff --git a/src/sentencepiece_processor.h b/src/sentencepiece_processor.h
index df1022a..213fdf8 100644
--- a/src/sentencepiece_processor.h
+++ b/src/sentencepiece_processor.h
@@ -92,10 +92,18 @@ class SentencePieceProcessor {
   // Returns false if |filename| cannot be loaded.
   virtual bool Load(const std::string &filename);
 
+  // Loads model from |istream|.
+  // Returns false if |istream| cannot be loaded.
+  virtual bool Load(std::istream &istream);
+
   // Loads model from |filename|.
   // Dies if |filename| cannot be loaded.
   virtual void LoadOrDie(const std::string &filename);
 
+  // Loads model from |istream|.
+  // Dies if |istream| cannot be loaded.
+  virtual void LoadOrDie(std::istream &istream);
+
   // Sets encode extra_option sequence.
   virtual void SetEncodeExtraOptions(const std::string &extra_option);