putting login back

model.py

@@ -10,12 +10,12 @@ model_id = 'meta-llama/Llama-2-13b-chat-hf'
 
 if torch.cuda.is_available():
     
-    # tok = os.environ['HF_TOKEN']
+    tok = os.environ['HF_TOKEN']
     
-    # login(new_session=True,
-    #       write_permission=False,
-    #      token=tok
-    #       )
+    login(new_session=True,
+          write_permission=False,
+         token=tok
+          )
     
     config = AutoConfig.from_pretrained(model_id,
         use_auth_token=True)
@@ -30,8 +30,7 @@ if torch.cuda.is_available():
     )
 else:
     model = None
-tokenizer = AutoTokenizer.from_pretrained(model_id,
-        use_auth_token=True)
+tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 
 def get_prompt(message: str, chat_history: list[tuple[str, str]],