Sebastiankay commited on
Commit
644675b
1 Parent(s): 5ea2e5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -234,7 +234,7 @@ with gr.Blocks(theme=theme, head=custom_head, css=custom_css, js=custom_js, titl
234
  inference_models = ["FLUX-API-OLD"]
235
  selected_model = "FLUX-API-OLD"
236
  models_new_api = ["FLUX", "FLUX-Realism", "FLUX-Pixel", "FLUX-Anime", "FLUX-3D", "FLUX-Disney", "ANY-DARK", "Stable-Diffusion-XL-Base", "Stable-Diffusion-XL-Lightning"]
237
- if check_api(API_V1) == 200:
238
  selected_model = "FLUX"
239
  for model in models_new_api:
240
  inference_models.append(model)
 
234
  inference_models = ["FLUX-API-OLD"]
235
  selected_model = "FLUX-API-OLD"
236
  models_new_api = ["FLUX", "FLUX-Realism", "FLUX-Pixel", "FLUX-Anime", "FLUX-3D", "FLUX-Disney", "ANY-DARK", "Stable-Diffusion-XL-Base", "Stable-Diffusion-XL-Lightning"]
237
+ if check_api(API_V1_IMAGE) == 200:
238
  selected_model = "FLUX"
239
  for model in models_new_api:
240
  inference_models.append(model)