AlStable commited on
Commit
4c5fefe
·
1 Parent(s): 060702a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -22,14 +22,14 @@ models = [
22
  Model("SD21","models/stabilityai/stable-diffusion-2-1", "")
23
  ]
24
 
25
- custom_model = "no style selected"
26
 
27
  def selectModel(message):
28
  message = message.lower()
29
  for i in range(len(models)):
30
  if message.find(models[i].prefix)!=-1:
31
  c_model=models[i].path
32
- return custom_model
33
  else:
34
  c_model=models[i].path
35
  return c_model
@@ -38,7 +38,7 @@ sandbox = gr.Interface.load(
38
  fn= selectModel,
39
  name= custom_model,
40
  title="""AlStable sandbox""",
41
- inputs = gr.Textbox(label="Prompt", show_label=False, max_lines=2, placeholder="Enter your prompt~", elem_id="input-prompt"),
42
  description="""Demo for <a href="https://huggingface.co/stabilityai/stable-diffusion-2-1">AlStable</a> Stable Diffusion model.""",
43
  article=article,
44
  api_key=API_KEY
 
22
  Model("SD21","models/stabilityai/stable-diffusion-2-1", "")
23
  ]
24
 
25
+ custom_model = "models/stabilityai/stable-diffusion-2-1"
26
 
27
  def selectModel(message):
28
  message = message.lower()
29
  for i in range(len(models)):
30
  if message.find(models[i].prefix)!=-1:
31
  c_model=models[i].path
32
+ return c_model
33
  else:
34
  c_model=models[i].path
35
  return c_model
 
38
  fn= selectModel,
39
  name= custom_model,
40
  title="""AlStable sandbox""",
41
+ inputs = gr.Textbox(label="Prompt", show_label=False, max_lines=2, placeholder="Enter your prompt", elem_id="input-prompt"),
42
  description="""Demo for <a href="https://huggingface.co/stabilityai/stable-diffusion-2-1">AlStable</a> Stable Diffusion model.""",
43
  article=article,
44
  api_key=API_KEY