diff --git a/senju/haiku.py b/senju/haiku.py index 128ba79..195ae76 100644 --- a/senju/haiku.py +++ b/senju/haiku.py @@ -109,7 +109,7 @@ class Haiku: :rtype: Haiku :raises: Exception if took the llm too many tries to write a fitting - haiku or JSONDecodeError if working with JSON failed + haiku or JSONDecodeError if working with JSON failed """ ai_gen_request = { "model": "haiku", diff --git a/senju/image_reco.py b/senju/image_reco.py index e4f45c9..4fdc093 100644 --- a/senju/image_reco.py +++ b/senju/image_reco.py @@ -1,6 +1,6 @@ """ Senju Image Recognition Module -============================= +============================== A module providing image description generation capabilities for the Senju haiku application. @@ -31,7 +31,7 @@ Dependencies * transformers: Hugging Face's library providing access to pre-trained models Implementation Details ---------------------- +---------------------- The module initializes a BLIP model (Bootstrapped Language-Image Pre-training) which can understand visual content and generate natural language descriptions. The implementation handles image loading, preprocessing, model inference,