Spaces:
Running
on
Zero
Running
on
Zero
DjStompzone
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -6,6 +6,11 @@ from diffusers.models.controlnet_flux import FluxControlNetModel
|
|
6 |
import random
|
7 |
import numpy as np
|
8 |
|
|
|
|
|
|
|
|
|
|
|
9 |
# Initialize models
|
10 |
base_model = 'black-forest-labs/FLUX.1-dev'
|
11 |
controlnet_model = 'promeai/FLUX.1-controlnet-lineart-promeai'
|
|
|
6 |
import random
|
7 |
import numpy as np
|
8 |
|
9 |
+
import os
|
10 |
+
from huggingface_hub import login
|
11 |
+
|
12 |
+
login(os.getenv("hfapikey"))
|
13 |
+
|
14 |
# Initialize models
|
15 |
base_model = 'black-forest-labs/FLUX.1-dev'
|
16 |
controlnet_model = 'promeai/FLUX.1-controlnet-lineart-promeai'
|