Baptlem commited on
Commit
a77d97b
1 Parent(s): 5c9fc3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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/baptlem-controlnet)
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/baptlem-controlnet",
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/baptlem-controlnet"
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/baptlem-controlnet")
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,