⚡ Support dotenv

MrGraversen · MrGraversen · commit e94771a46019 · 2024-11-01T11:10:51.000+01:00
diff --git a/replicate-tools/.gitignore b/replicate-tools/.gitignore
@@ -1 +1,2 @@
-schemas
+schemas
+.env
diff --git a/replicate-tools/default_replicate_models.txt b/replicate-tools/default_replicate_models.txt
@@ -1,3 +1,4 @@
+meta/meta-llama-3.1-405b-instruct
 meta/meta-llama-3-70b-instruct
 meta/meta-llama-3-8b-instruct
 thomasmol/whisper-diarization
diff --git a/replicate-tools/download_replicate_schemas.py b/replicate-tools/download_replicate_schemas.py
@@ -4,6 +4,9 @@
 import shutil
 
 import requests
+from dotenv import load_dotenv
+
+load_dotenv()
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
diff --git a/replicate-tools/requirements.txt b/replicate-tools/requirements.txt
@@ -1 +1,2 @@
-requests
+requests
+python-dotenv

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
	`1`	`+meta/meta-llama-3.1-405b-instruct`
`1`	`2`	`meta/meta-llama-3-70b-instruct`
`2`	`3`	`meta/meta-llama-3-8b-instruct`
`3`	`4`	`thomasmol/whisper-diarization`