add login using script

SukhmeetSingh2002 · Oct 1, 2023 · df34032 · df34032
1 parent 1da8844
commit df34032
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/bot.py b/src/bot.py
@@ -2,9 +2,13 @@
 from transformers import pipeline
 import transformers
 import torch
+from huggingface_hub import login
+
+import os
 
 class LlamaBot:
  def __init__(self):
+ login(token = os.environ['HF_TOKEN'])
  self.model = "meta-llama/Llama-2-7b-chat-hf"
  self.tokenizer = AutoTokenizer.from_pretrained(self.model, use_auth_token=True)
  self.llama_pipeline = pipeline(