sheami / common.py
vikramvasudevan's picture
Upload folder using huggingface_hub
b1447d4 verified
import os
from dotenv import load_dotenv
def get_db():
load_dotenv(override=True)
DB_URI = os.getenv("MONGODB_URI")
DB_NAME = os.getenv("MONGODB_DB", "sheami")
from modules.db import SheamiDB # imported here so this file stays portable
return SheamiDB(DB_URI, db_name=DB_NAME)