AlStable commited on
Commit
4475ccb
·
1 Parent(s): 6757c45

Update app.py

Browse files

dynamic model switching

Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -17,10 +17,11 @@ class Model:
17
  models = [
18
  Model("Marvel","ItsJayQz/Marvel_WhatIf_Diffusion", "Marvel"),
19
  Model("Cyberpunk Anime Diffusion", "DGSpitzer/Cyberpunk-Anime-Diffusion", "dgs illustration style "),
20
- Model("Guan Yu Diffusion", "DGSpitzer/Guan-Yu-Diffusion", "Guan-Yu")
 
21
  ]
22
 
23
- custom_model = "models/stabilityai/stable-diffusion-2-1"
24
 
25
  gr.Interface.load(
26
  name= custom_model,
 
17
  models = [
18
  Model("Marvel","ItsJayQz/Marvel_WhatIf_Diffusion", "Marvel"),
19
  Model("Cyberpunk Anime Diffusion", "DGSpitzer/Cyberpunk-Anime-Diffusion", "dgs illustration style "),
20
+ Model("Guan Yu Diffusion", "DGSpitzer/Guan-Yu-Diffusion", "Guan-Yu"),
21
+ Model("SD21","models/stabilityai/stable-diffusion-2-1", "")
22
  ]
23
 
24
+ custom_model = models[3].path
25
 
26
  gr.Interface.load(
27
  name= custom_model,