Update model.py

model.py

@@ -4,7 +4,7 @@ from typing import Iterator
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 
-model_id = 'meta-llama/Llama-2-7b-chat-hf'
+model_id = 'LinkSoul/Chinese-Llama-2-7b'
 
 if torch.cuda.is_available():
     model = AutoModelForCausalLM.from_pretrained(
@@ -17,6 +17,7 @@ else:
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 
+
 def get_prompt(message: str, chat_history: list[tuple[str, str]],
                system_prompt: str) -> str:
     texts = [f'[INST] <<SYS>>\n{system_prompt}\n<</SYS>>\n\n']