diff --git a/llama_hub/llama_packs/llava_completion/base.py b/llama_hub/llama_packs/llava_completion/base.py index d20864f1dc..cde63bf43c 100644 --- a/llama_hub/llama_packs/llava_completion/base.py +++ b/llama_hub/llama_packs/llava_completion/base.py @@ -17,8 +17,8 @@ def __init__( """Init params.""" import os - if os.environ["REPLICATE_API_TOKEN"] == "": - raise ValueError("Replicate API Token is missing.") + if not os.environ.get("REPLICATE_API_TOKEN", None): + raise ValueError("Replicate API Token is missing or blank.") self.image_url = image_url