nbroad commited on
Commit
308b3c4
·
verified ·
1 Parent(s): 8d5f206

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +7 -0
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")