Update app.py

app.py

@@ -4,7 +4,7 @@ import collections
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Define model name - use a very small model
-MODEL_NAME = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"  # Chat-tuned 1.1B parameter model
+MODEL_NAME = "Qwen/Qwen2-0.5B-Instruct"  # 0.5B parameter instruction model
 
 print(f"Loading model {MODEL_NAME}...")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)