Update app.py

app.py

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