Update app.py

app.py

@@ -8,14 +8,14 @@ import random
 from datasets import load_dataset
 
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
-MODEL_ID = "Aratako/sarashina2.1-1b-sft"
+MODEL_ID = "CohereForAI/c4ai-command-r7b-12-2024"
 MODELS = os.environ.get("MODELS")
 MODEL_NAME = MODEL_ID.split("/")[-1]
 
 TITLE = "<h1><center>New japanese LLM model webui</center></h1>"
 
 DESCRIPTION = f"""
-<h3>MODEL: <a href="https://huggingface.co/Aratako/sarashina2.1-1b-sft">Aratako/sarashina2.1-1b-sft</a></h3>
+<h3>MODEL: <a href="https://huggingface.co/CohereForAI/c4ai-command-r7b-12-2024">CohereForAI/c4ai-command-r7b-12-2024</a></h3>
 <center>
 <p>
 <br>
@@ -85,7 +85,7 @@ def stream_chat(message: str, history: list, temperature: float, max_new_tokens:
         max_new_tokens=max_new_tokens, 
         do_sample=True, 
         temperature=temperature,
-        eos_token_id=[2, 102401],
+        #eos_token_id=[255001],
     )
     
     thread = Thread(target=model.generate, kwargs=generate_kwargs)