Avijit Ghosh commited on
Commit
8a06c57
1 Parent(s): 31a0f6f

add dropdown

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -47,15 +47,15 @@ def load_model(model_name):
47
  raise ValueError("Unknown model name")
48
  return pipeline
49
 
50
- choices=[
51
- "stabilityai/sdxl-turbo",
52
- "runwayml/stable-diffusion-v1-5",
53
- "ByteDance/SDXL-Lightning",
54
- "segmind/SSD-1B"
55
- ]
56
 
57
- for model_name in choices:
58
- load_model(model_name)
59
 
60
  # Initialize the default model
61
  default_model = "stabilityai/sdxl-turbo"
 
47
  raise ValueError("Unknown model name")
48
  return pipeline
49
 
50
+ # choices=[
51
+ # "stabilityai/sdxl-turbo",
52
+ # "runwayml/stable-diffusion-v1-5",
53
+ # "ByteDance/SDXL-Lightning",
54
+ # "segmind/SSD-1B"
55
+ # ]
56
 
57
+ # for model_name in choices:
58
+ # load_model(model_name)
59
 
60
  # Initialize the default model
61
  default_model = "stabilityai/sdxl-turbo"