evijit HF staff commited on
Commit
1ff1569
·
verified ·
1 Parent(s): 6704535

Update app.py

Browse files

Hiding flux for now because of hardware restrictions

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -72,7 +72,7 @@ def load_model(model_name):
72
  return pipeline
73
 
74
  # Initialize the default model
75
- default_model = "black-forest-labs/FLUX.1-dev"
76
  pipeline_text2image = load_model(default_model)
77
 
78
  @spaces.GPU
@@ -225,7 +225,7 @@ This demo provides an insightful look into how current text-to-image models hand
225
  model_dropdown = gr.Dropdown(
226
  label="Choose a model",
227
  choices=[
228
- "black-forest-labs/FLUX.1-dev",
229
  "stabilityai/stable-diffusion-3-medium-diffusers",
230
  "stabilityai/sdxl-turbo",
231
  "ByteDance/SDXL-Lightning",
 
72
  return pipeline
73
 
74
  # Initialize the default model
75
+ default_model = "stabilityai/stable-diffusion-xl-base-1.0"
76
  pipeline_text2image = load_model(default_model)
77
 
78
  @spaces.GPU
 
225
  model_dropdown = gr.Dropdown(
226
  label="Choose a model",
227
  choices=[
228
+ # "black-forest-labs/FLUX.1-dev",
229
  "stabilityai/stable-diffusion-3-medium-diffusers",
230
  "stabilityai/sdxl-turbo",
231
  "ByteDance/SDXL-Lightning",