Update src/app/model.py

src/app/model.py

@@ -18,7 +18,7 @@ load_dotenv()
 access_token = os.environ.get("ACCESS_TOKEN")
 
 
-def load_model_and_tokenizer(model_name: str, device: str) -> Any:
+def load_model_tokenizer_and_processor(model_name: str, device: str) -> Any:
     """
     Load the model, tokenizer and processor.
 
@@ -49,13 +49,13 @@ def load_model_and_tokenizer(model_name: str, device: str) -> Any:
         )
         model.eval()
 
-        # Log the successful loading of the model and tokenizer
-        logging.info("Model and tokenizer loaded successfully.")
+        # Log the successful loading of the model, tokenizer and processor
+        logging.info("Model, tokenizer and processor loaded successfully.")
 
         # Return the model, tokenizer and processor
         return model, tokenizer, processor
 
-    # Handle exceptions that may occur during model and tokenizer loading
+    # Handle exceptions that may occur during model, tokenizer and processor loading
     except Exception as e:
         # Custom exception handling
         raise CustomExceptionHandling(e, sys) from e