Update app.py

app.py

@@ -1,16 +1,19 @@
 import gradio as gr
+# Use a pipeline as a high-level helper
+
 # Use a pipeline as a high-level helper
 from transformers import pipeline
 
-pipe = pipeline("text-generation", model="tiiuae/falcon-7b", trust_remote_code=True)
-# Load model directly
-# Load model directly
-# Load model directly
+messages = [
+    {"role": "user", "content": "Who are you?"},
+]
+pipe = pipeline("text-generation", model="Qwen/Qwen2.5-1.5B")
+pipe(messages)
 # Load model directly
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-7b", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-7b", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-1.5B")
+model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-1.5B")
 def respond(
     message,
     history: list[tuple[str, str]],