Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
@@ -20,6 +20,13 @@ from contextlib import asynccontextmanager
|
|
20 |
logging.basicConfig(level=logging.INFO)
|
21 |
logger = logging.getLogger(__name__)
|
22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
23 |
# Global variables for dataset management
|
24 |
DATASET_REPO_NAME = os.getenv("DATASET_REPO_NAME", "nbroad/hf-inference-providers-data")
|
25 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
|
|
20 |
logging.basicConfig(level=logging.INFO)
|
21 |
logger = logging.getLogger(__name__)
|
22 |
|
23 |
+
|
24 |
+
IN_SPACE = os.getenv("SPACE_REPO_NAME") is not None
|
25 |
+
|
26 |
+
if not IN_SPACE:
|
27 |
+
from dotenv import load_dotenv
|
28 |
+
load_dotenv()
|
29 |
+
|
30 |
# Global variables for dataset management
|
31 |
DATASET_REPO_NAME = os.getenv("DATASET_REPO_NAME", "nbroad/hf-inference-providers-data")
|
32 |
HF_TOKEN = os.getenv("HF_TOKEN")
|