AlStable commited on
Commit
f25ca48
·
1 Parent(s): 1a88a05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -25,7 +25,7 @@ models = [
25
  Model("SD21","models/stabilityai/stable-diffusion-2-1", "")
26
  ]
27
 
28
- """custom_model = models[3].path"""
29
 
30
  def selectModel(message):
31
  message = message.lower()
@@ -38,8 +38,8 @@ def selectModel(message):
38
  return c_model
39
 
40
  sandbox = gr.Interface.load(
41
- name=custom_model,
42
  fn= selectModel,
 
43
  title="""AlStable sandbox""",
44
  inputs = gr.Textbox(label="Prompt", show_label=False, max_lines=2, placeholder="Enter your prompt~", elem_id="input-prompt"),
45
  description="""Demo for <a href="https://huggingface.co/stabilityai/stable-diffusion-2-1">AlStable</a> Stable Diffusion model.""",
@@ -47,6 +47,6 @@ sandbox = gr.Interface.load(
47
  api_key=API_KEY
48
  )
49
 
50
- """sandbox.custom_model=sandbox.selectModel(inputs)"""
51
 
52
  sandbox.queue(concurrency_count=20).launch()
 
25
  Model("SD21","models/stabilityai/stable-diffusion-2-1", "")
26
  ]
27
 
28
+ custom_model = models[3].path
29
 
30
  def selectModel(message):
31
  message = message.lower()
 
38
  return c_model
39
 
40
  sandbox = gr.Interface.load(
 
41
  fn= selectModel,
42
+ name= custom_model,
43
  title="""AlStable sandbox""",
44
  inputs = gr.Textbox(label="Prompt", show_label=False, max_lines=2, placeholder="Enter your prompt~", elem_id="input-prompt"),
45
  description="""Demo for <a href="https://huggingface.co/stabilityai/stable-diffusion-2-1">AlStable</a> Stable Diffusion model.""",
 
47
  api_key=API_KEY
48
  )
49
 
50
+ sandbox.custom_model=sandbox.selectModel(inputs)
51
 
52
  sandbox.queue(concurrency_count=20).launch()