AlStable commited on
Commit
601c77e
·
1 Parent(s): d46f263

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -25,10 +25,11 @@ models = [
25
  Model("Portrait plus", "models/wavymulder/portraitplus", "portrait+ style"),
26
  Model("classic Disney", "models/nitrosocke/classic-anim-diffusion", "classic disney style"),
27
  Model("vintedois", "models/22h/vintedois-diffusion-v0-1", "vintedois style"),
 
28
  Model("SD21","models/stabilityai/stable-diffusion-2-1", "sd21 default style")
29
  ]
30
 
31
- custom_model = "models/stabilityai/stable-diffusion-2-1"
32
 
33
  def selectModel(message):
34
  message = message.lower()
@@ -51,8 +52,6 @@ sandbox = gr.Interface.load(
51
  api_key=API_KEY
52
  )
53
 
54
- """"logging.warning('prompt => '+sandbox.inputs)"""
55
-
56
  logging.warning('model chosen = '+custom_model)
57
 
58
  sandbox.queue(concurrency_count=20).launch()
 
25
  Model("Portrait plus", "models/wavymulder/portraitplus", "portrait+ style"),
26
  Model("classic Disney", "models/nitrosocke/classic-anim-diffusion", "classic disney style"),
27
  Model("vintedois", "models/22h/vintedois-diffusion-v0-1", "vintedois style"),
28
+ Model("dreamlike", "models/dreamlike-art/dreamlike-diffusion-1.0","dreamlike style"),
29
  Model("SD21","models/stabilityai/stable-diffusion-2-1", "sd21 default style")
30
  ]
31
 
32
+ custom_model = "models/dreamlike-art/dreamlike-diffusion-1.0"
33
 
34
  def selectModel(message):
35
  message = message.lower()
 
52
  api_key=API_KEY
53
  )
54
 
 
 
55
  logging.warning('model chosen = '+custom_model)
56
 
57
  sandbox.queue(concurrency_count=20).launch()