ragunath-ravi commited on
Commit
33adcce
·
verified ·
1 Parent(s): 2ea78e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -28,7 +28,13 @@ logging.basicConfig(level=logging.INFO)
28
  logger = logging.getLogger(__name__)
29
 
30
  # Get HF token from environment
31
- HF_TOKEN = os.getenv('HF_TOKEN')
 
 
 
 
 
 
32
 
33
  # MCP Message Structure
34
  @dataclass
 
28
  logger = logging.getLogger(__name__)
29
 
30
  # Get HF token from environment
31
+ HF_TOKEN = os.getenv('hf_token')
32
+
33
+ if HF_TOKEN is None:
34
+ logger.error("FATAL ERROR: HuggingFace token (HF_TOKEN) environment variable is not set.")
35
+ logger.error("Please set the HF_TOKEN environment variable before running the application.")
36
+ sys.exit("HuggingFace token (HF_TOKEN) is not set. Exiting.")
37
+
38
 
39
  # MCP Message Structure
40
  @dataclass