Update app.py

app.py

@@ -36,7 +36,7 @@ def initialize_model():
     global pipeline_instance
     try:
         logger.info("Initializing the base model and tokenizer.")
-        base_model_repo = "Yhhxhfh/test"
+        base_model_repo = "meta-llama/Llama-3.2-1B"
         model = AutoModelForCausalLM.from_pretrained(base_model_repo, cache_dir=CACHE_DIR)
         tokenizer = AutoTokenizer.from_pretrained(base_model_repo, cache_dir=CACHE_DIR)
         if tokenizer.pad_token is None: