Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
 import gradio as gr
 from threading import Thread
 
-model_id = "sudy-super/Yamase-12B"
+model_id = "llm-jp/llm-jp-3-1.8b-instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(
   model_id,
@@ -15,10 +15,10 @@ model = AutoModelForCausalLM.from_pretrained(
   use_flash_attention_2=True,
 )
 
-TITLE = "<h1><center>sudy-super/Yamase-12B Chat webui</center></h1>"
+TITLE = "<h1><center>llm-jp/llm-jp-3-1.8b-instruct Chat webui</center></h1>"
 
 DESCRIPTION = """
-<h3>MODEL: <a href="https://huggingface.co/sudy-super/Yamase-12B">sudy-super/Yamase-12B</a></h3>
+<h3>MODEL: <a href="https://huggingface.co/llm-jp/llm-jp-3-1.8b-instruct">llm-jp/llm-jp-3-1.8b-instruct</a></h3>
 <center>
 <p>This model is designed for conversational interactions.</p>
 </center>