Spaces:
Running
Running
Update app.py
Browse files
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 |
-
"
|
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
|
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(
|