Update Dockerfile

Dockerfile

@@ -47,15 +47,13 @@ from sentence_transformers import SentenceTransformer
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-# Download and save sentence transformer model
 print("Downloading sentence transformer model...")
 model = SentenceTransformer("sentence-transformers/all-mpnet-base-v2")
 model.save("/app/models/sentence_transformer")
-print("Sentence transformer model saved successfully!")
+print("Sentence transformer model saved!")
 
-# Download and save Qwen model and tokenizer
-print("Downloading Qwen model and tokenizer...")
-model_name = "Qwen/Qwen3.5-0.8B-Instruct"
+print("Downloading Qwen model...")
+model_name = "Qwen/Qwen2.5-0.5B-Instruct"  # Fixed: was Qwen3.5 (doesn't exist)
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
@@ -63,10 +61,9 @@ model = AutoModelForCausalLM.from_pretrained(
     torch_dtype=torch.float16,
     device_map=None
 )
-
 tokenizer.save_pretrained("/app/models/qwen")
 model.save_pretrained("/app/models/qwen")
-print("Qwen model and tokenizer saved successfully!")
+print("Qwen model saved!")
 EOF
 
 # Download models during build