AlStable commited on
Commit
cde57f0
·
1 Parent(s): 57e33eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -19,12 +19,13 @@ AssertionError: parameter: src must be one of dict_keys(['huggingface', 'models'
19
  """
20
  models = [
21
  Model("Marvel","models/ItsJayQz/Marvel_WhatIf_Diffusion", ""),
22
- Model("Cyberpunk Anime Diffusion", "models/DGSpitzer/Cyberpunk-Anime-Diffusion", "dgs illustration style"),
23
- Model("Guan Yu Diffusion", "models/DGSpitzer/Guan-Yu-Diffusion", "Guan-Yu style"),
 
24
  Model("SD21","models/stabilityai/stable-diffusion-2-1", "")
25
  ]
26
 
27
- custom_model = models[0].path
28
 
29
  gr.Interface.load(
30
  name= custom_model,
 
19
  """
20
  models = [
21
  Model("Marvel","models/ItsJayQz/Marvel_WhatIf_Diffusion", ""),
22
+ Model("Cyberpunk Anime Diffusion", "models/DGSpitzer/Cyberpunk-Anime-Diffusion", "dgs illustration style "),
23
+ Model("Guan Yu Diffusion", "models/DGSpitzer/Guan-Yu-Diffusion", "Guan-Yu"),
24
+ Model("classic Disney", "models/nitrosocke/classic-anim-diffusion", "classic disney style"),
25
  Model("SD21","models/stabilityai/stable-diffusion-2-1", "")
26
  ]
27
 
28
+ custom_model = models[3].path
29
 
30
  gr.Interface.load(
31
  name= custom_model,