model change

ChatController.py

@@ -9,7 +9,7 @@ app = Flask(__name__)
 CORS(app)
 
 chatService = ChatService()
-chatService.load_model("collinear-ai/Llama-2-7b-chat-crs")
+chatService.load_model("collinear-ai/LLaMA-2-7B-chat-csr")
 
 @app.route("/chat", methods=['POST'])
 @cross_origin(origin='*')

ChatService.py

@@ -1,7 +1,7 @@
 import os
 import torch
 import transformers
-# from huggingface_hub import login
+from huggingface_hub import login
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 
@@ -19,7 +19,7 @@ class ChatService:
         gpu_count = torch.cuda.device_count()
         print('gpu_count', gpu_count)
 
-        # login("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")
+        login("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")
 
         tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
         pipeline = transformers.pipeline(