Update README.md

README.md

@@ -23,7 +23,7 @@ import torch
 from hindi_embeddings import SentencePieceTokenizerWrapper
 from convaicausallm_model import ConvaiCausalLM, ConvaiCausalLMConfig
 from safetensors.torch import load_file
-
+import os 
 class HindiLLMGenerator:
     def __init__(self, model_path, device=None):
         # Set device