Update base.py

logan-markewich · Nov 16, 2023 · 095e345 · 095e345
1 parent 9e8fa67
commit 095e345
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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