Update handler.py
Browse files- 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
|
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(
|