Refactor model loading to use a consistent model name variable in PoetryGenerationService

app/services/poetry_generation.py

@@ -23,10 +23,10 @@ class ModelManager:
     def __init__(self):
             
             # Initialize tokenizer and model
-            self.tokenizer = AutoTokenizer.from_pretrained(self._model_name)
+            self.tokenizer = AutoTokenizer.from_pretrained(model_name)
             self.tokenizer.pad_token = self.tokenizer.eos_token
             self.model = AutoModelForCausalLM.from_pretrained(
-                self._model_name,
+                model_name,
                 torch_dtype=torch.float16,
                 device_map="auto"
             )
@@ -54,7 +54,7 @@ def get_hf_token() -> str:
             "Please set your Hugging Face access token."
         )
     return token
-
+model_name = "meta-llama/Llama-3.2-1B-Instruct"
 class PoetryGenerationService:
     def __init__(self):
         # Get model manager instance
@@ -66,11 +66,11 @@ class PoetryGenerationService:
         """Preload the models during application startup"""
         try:
             # Initialize tokenizer and model
-            self.tokenizer = AutoTokenizer.from_pretrained(self._model_name)
+            self.tokenizer = AutoTokenizer.from_pretrained(model_name)
             self.tokenizer.pad_token = self.tokenizer.eos_token
 
             self.model = AutoModelForCausalLM.from_pretrained(
-                self._model_name,
+                model_name,
                 torch_dtype=torch.float16,
                 device_map="auto"
             )