OMilosh commited on
Commit
b25f434
·
verified ·
1 Parent(s): e390658

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import gradio as gr
2
  import numpy as np
3
  import random
4
- from tqdm import tqdm
5
 
6
  # import spaces #[uncomment to use ZeroGPU]
7
  from diffusers import DiffusionPipeline
@@ -14,9 +13,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
14
 
15
  available_models = [
16
  "stabilityai/sdxl-turbo",
17
- "CompVis/stable-diffusion-v1-4",
18
- "runwayml/stable-diffusion-v1-5",
19
- ]
20
 
21
  MAX_SEED = np.iinfo(np.int32).max
22
  MAX_IMAGE_SIZE = 1024
@@ -153,7 +150,7 @@ with gr.Blocks(css=css) as demo:
153
 
154
  loaded_models = {}
155
 
156
- for model in tqdm(available_models):
157
  loaded_models[model] = init_model(model)
158
 
159
  gr.on(
 
1
  import gradio as gr
2
  import numpy as np
3
  import random
 
4
 
5
  # import spaces #[uncomment to use ZeroGPU]
6
  from diffusers import DiffusionPipeline
 
13
 
14
  available_models = [
15
  "stabilityai/sdxl-turbo",
16
+ "stabilityai/sd-turbo"]
 
 
17
 
18
  MAX_SEED = np.iinfo(np.int32).max
19
  MAX_IMAGE_SIZE = 1024
 
150
 
151
  loaded_models = {}
152
 
153
+ for model in available_models:
154
  loaded_models[model] = init_model(model)
155
 
156
  gr.on(