Update main_process/salamandra_router.py
Browse files
main_process/salamandra_router.py
CHANGED
|
@@ -44,7 +44,7 @@ EMBEDDINGS_ROOT = Path("/data/embeddings")
|
|
| 44 |
MEDIA_ROOT = Path("/data/media")
|
| 45 |
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
|
| 46 |
router = APIRouter(prefix="/salamandra", tags=["Salamandra Process"])
|
| 47 |
-
HF_TOKEN = os.getenv("
|
| 48 |
OPEN_AI_KEY = os.getenv("OPEN_AI_KEY")
|
| 49 |
|
| 50 |
class DataHub:
|
|
|
|
| 44 |
MEDIA_ROOT = Path("/data/media")
|
| 45 |
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
|
| 46 |
router = APIRouter(prefix="/salamandra", tags=["Salamandra Process"])
|
| 47 |
+
HF_TOKEN = os.getenv("SALAMANDRA_TK")
|
| 48 |
OPEN_AI_KEY = os.getenv("OPEN_AI_KEY")
|
| 49 |
|
| 50 |
class DataHub:
|