Update app.py
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ We then trained four Control-Nets with different mixtures of 2 datasets (Coyo-70
|
|
41 |
|
42 |
Model Description and Training Process: Please refer to the readme file attached to the model repository.
|
43 |
|
44 |
-
Model Repository: [ControlNet repo](https://huggingface.co/Baptlem/
|
45 |
|
46 |
"""
|
47 |
|
@@ -74,7 +74,7 @@ def create_key(seed=0):
|
|
74 |
|
75 |
def load_controlnet(controlnet_version):
|
76 |
controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
|
77 |
-
"Baptlem/
|
78 |
subfolder=controlnet_version,
|
79 |
from_flax=True,
|
80 |
dtype=jnp.float32,
|
@@ -104,7 +104,7 @@ def load_sb_pipe(controlnet_version, sb_path="runwayml/stable-diffusion-v1-5"):
|
|
104 |
|
105 |
|
106 |
|
107 |
-
controlnet_path = "Baptlem/
|
108 |
controlnet_version = "coyo-500k"
|
109 |
|
110 |
# Constants
|
@@ -252,7 +252,7 @@ def create_demo(process, max_images=12, default_num_images=4):
|
|
252 |
model = gr.Dropdown(choices=["coyo-500k", "bridge-2M", "coyo1M-bridge2M", "coyo2M-bridge325k"],
|
253 |
value="coyo-500k",
|
254 |
label="Model used for inference",
|
255 |
-
info="Find every models at https://huggingface.co/Baptlem/
|
256 |
seed = gr.Slider(label='Seed',
|
257 |
minimum=-1,
|
258 |
maximum=2147483647,
|
|
|
41 |
|
42 |
Model Description and Training Process: Please refer to the readme file attached to the model repository.
|
43 |
|
44 |
+
Model Repository: [ControlNet repo](https://huggingface.co/Baptlem/UCDR-Net_models)
|
45 |
|
46 |
"""
|
47 |
|
|
|
74 |
|
75 |
def load_controlnet(controlnet_version):
|
76 |
controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
|
77 |
+
"Baptlem/UCDR-Net_models",
|
78 |
subfolder=controlnet_version,
|
79 |
from_flax=True,
|
80 |
dtype=jnp.float32,
|
|
|
104 |
|
105 |
|
106 |
|
107 |
+
controlnet_path = "Baptlem/UCDR-Net_models"
|
108 |
controlnet_version = "coyo-500k"
|
109 |
|
110 |
# Constants
|
|
|
252 |
model = gr.Dropdown(choices=["coyo-500k", "bridge-2M", "coyo1M-bridge2M", "coyo2M-bridge325k"],
|
253 |
value="coyo-500k",
|
254 |
label="Model used for inference",
|
255 |
+
info="Find every models at https://huggingface.co/Baptlem/UCDR-Net_models")
|
256 |
seed = gr.Slider(label='Seed',
|
257 |
minimum=-1,
|
258 |
maximum=2147483647,
|