Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import torch
 
 # Load model and tokenizer
-model_name = "LiquidAI/LFM2-350M"
+model_name = "baidu/ERNIE-4.5-0.3B-PT"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
@@ -51,9 +51,9 @@ def chat_function(message, history):
     return response
 
 # Gradio Interface
-with gr.Blocks(title="LFM2-350M Chat") as demo:
-    gr.Markdown("# 🤖 LFM2-350M Simple Chat")
-    gr.Markdown("A minimal chat interface using `LiquidAI/LFM2-350M`. Optimized for clean single-turn responses.")
+with gr.Blocks(title="baidu/ERNIE-4.5-0.3B-PT Chat") as demo:
+    gr.Markdown("# 🤖 baidu/ERNIE-4.5-0.3B-PT Simple Chat")
+    gr.Markdown("A minimal chat interface using `baidu/ERNIE-4.5-0.3B-PT`. Optimized for clean single-turn responses.")
 
     chatbot = gr.Chatbot(height=400)
     msg = gr.Textbox(label="Type your message", placeholder="Say something...")