Jackflack09 commited on
Commit
f5a07f3
·
1 Parent(s): 5d139de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -48,13 +48,10 @@ models = [
48
  Model("TrinArt v2", "naclbit/trinart_stable_diffusion_v2"),
49
  Model("Spider-Verse", "nitrosocke/spider-verse-diffusion", "spiderverse style "),
50
  Model("Robo Diffusion", "nousr/robo-diffusion"),
 
51
  Model("Epic Diffusion", "johnslegers/epic-diffusion")
52
  ]
53
 
54
- custom_model = None
55
- if is_colab:
56
- models.insert(0, Model("Custom model"))
57
- custom_model = models[0]
58
 
59
  last_mode = "txt2img"
60
  current_model = models[1] if is_colab else models[0]
 
48
  Model("TrinArt v2", "naclbit/trinart_stable_diffusion_v2"),
49
  Model("Spider-Verse", "nitrosocke/spider-verse-diffusion", "spiderverse style "),
50
  Model("Robo Diffusion", "nousr/robo-diffusion"),
51
+ Model("U Pron", "stablediffusionapi/urpm", "pron""),
52
  Model("Epic Diffusion", "johnslegers/epic-diffusion")
53
  ]
54
 
 
 
 
 
55
 
56
  last_mode = "txt2img"
57
  current_model = models[1] if is_colab else models[0]