SemaSci commited on
Commit
f8f03f7
·
verified ·
1 Parent(s): c6fdf73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -9,6 +9,7 @@ import torch
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
  #model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like to use
11
  model_repo_id = "CompVis/stable-diffusion-v1-4"
 
12
 
13
  if torch.cuda.is_available():
14
  torch_dtype = torch.float16
@@ -83,13 +84,21 @@ with gr.Blocks(css=css) as demo:
83
  result = gr.Image(label="Result", show_label=False)
84
 
85
  with gr.Accordion("Advanced Settings", open=False):
86
- model_repo_id = gr.Text(
 
 
 
 
 
 
 
87
  label="Model Id",
 
88
  max_lines=1,
89
  placeholder="Choose model",
90
  visible=True,
91
  value=model_repo_id,
92
- )
93
 
94
  negative_prompt = gr.Text(
95
  label="Negative prompt",
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
  #model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like to use
11
  model_repo_id = "CompVis/stable-diffusion-v1-4"
12
+ model_dropdown = ['stabilityai/sdxl-turbo', 'CompVis/stable-diffusion-v1-4' ]
13
 
14
  if torch.cuda.is_available():
15
  torch_dtype = torch.float16
 
84
  result = gr.Image(label="Result", show_label=False)
85
 
86
  with gr.Accordion("Advanced Settings", open=False):
87
+ # model_repo_id = gr.Text(
88
+ # label="Model Id",
89
+ # max_lines=1,
90
+ # placeholder="Choose model",
91
+ # visible=True,
92
+ # value=model_repo_id,
93
+ # )
94
+ model_repo_id = gr.DropDown(
95
  label="Model Id",
96
+ choices=model_dropdown,
97
  max_lines=1,
98
  placeholder="Choose model",
99
  visible=True,
100
  value=model_repo_id,
101
+ )
102
 
103
  negative_prompt = gr.Text(
104
  label="Negative prompt",