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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,12 +19,12 @@ 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"),
24
  Model("SD21","models/stabilityai/stable-diffusion-2-1", "")
25
  ]
26
 
27
- custom_model = models[2].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 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,