Bordoglor commited on
Commit
e0f53b6
·
verified ·
1 Parent(s): a6bebb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -20,11 +20,6 @@ pipe = pipe.to(device)
20
  MAX_SEED = np.iinfo(np.int32).max
21
  MAX_IMAGE_SIZE = 1024
22
 
23
- MODELS = [
24
- "stabilityai/sdxl-turbo",
25
- "CompVis/stable-diffusion-v1-4"
26
- ]
27
-
28
 
29
  # @spaces.GPU #[uncomment to use ZeroGPU]
30
  def infer(
@@ -77,7 +72,7 @@ with gr.Blocks(css=css) as demo:
77
 
78
  with gr.Row():
79
  model_dropdown = gr.Dropdown(
80
- choices=MODELS,
81
  value=model_repo_id,
82
  label="Model",
83
  info="Choose which diffusion model to use"
@@ -100,6 +95,7 @@ with gr.Blocks(css=css) as demo:
100
  with gr.Accordion("Advanced Settings", open=False):
101
  negative_prompt = gr.Text(
102
  label="Negative prompt",
 
103
  max_lines=1,
104
  placeholder="Enter a negative prompt",
105
  visible=False,
 
20
  MAX_SEED = np.iinfo(np.int32).max
21
  MAX_IMAGE_SIZE = 1024
22
 
 
 
 
 
 
23
 
24
  # @spaces.GPU #[uncomment to use ZeroGPU]
25
  def infer(
 
72
 
73
  with gr.Row():
74
  model_dropdown = gr.Dropdown(
75
+ choices=["stabilityai/sdxl-turbo", "CompVis/stable-diffusion-v1-4"],
76
  value=model_repo_id,
77
  label="Model",
78
  info="Choose which diffusion model to use"
 
95
  with gr.Accordion("Advanced Settings", open=False):
96
  negative_prompt = gr.Text(
97
  label="Negative prompt",
98
+ value="dog, cat",
99
  max_lines=1,
100
  placeholder="Enter a negative prompt",
101
  visible=False,