huyai123 commited on
Commit
27e6396
·
verified ·
1 Parent(s): 7754b09

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -11,7 +11,7 @@ class EndpointHandler:
11
  # Access the environment variable
12
  HUGGINGFACE_API_TOKEN = os.getenv('HUGGINGFACE_API_TOKEN')
13
  if not HUGGINGFACE_API_TOKEN:
14
- raise ValueError("HUGGINGFACE_API_TOKEN environment variable is not set")
15
 
16
  # Load model and pipeline
17
  self.controlnet = FluxControlNetModel.from_pretrained(
 
11
  # Access the environment variable
12
  HUGGINGFACE_API_TOKEN = os.getenv('HUGGINGFACE_API_TOKEN')
13
  if not HUGGINGFACE_API_TOKEN:
14
+ raise ValueError("HUGGINGFACE_API_TOKEN")
15
 
16
  # Load model and pipeline
17
  self.controlnet = FluxControlNetModel.from_pretrained(