Update app.py

app.py

@@ -3,8 +3,8 @@ import threading
 from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig, TextIteratorStreamer
 
 # Load the model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("SmallDoge/Doge-20M-Instruct")
-model = AutoModelForCausalLM.from_pretrained("SmallDoge/Doge-20M-Instruct", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("SmallDoge/Doge-20M-checkpoint")
+model = AutoModelForCausalLM.from_pretrained("SmallDoge/Doge-20M-checkpoint", trust_remote_code=True)
 
 # Generation configuration
 generation_config = GenerationConfig(
@@ -73,7 +73,7 @@ def chat(user_input, history):
 
 # Build the Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("## Chat with SmallDoge/Doge-20M-Instruct")
+    gr.Markdown("## Chat with SmallDoge/Doge-20M-checkpoint")
     chatbot = gr.Chatbot()  # displays the conversation as a list of (user, assistant) pairs
     with gr.Row():
         msg = gr.Textbox(show_label=False, placeholder="Type your message here...")