Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,15 +4,19 @@ import gradio as gr
|
|
4 |
import accelerate
|
5 |
|
6 |
models=[
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
|
|
14 |
]
|
15 |
|
|
|
|
|
|
|
16 |
def TextToImage(Prompt,model):
|
17 |
model_id = model
|
18 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.get_default_dtype())
|
|
|
4 |
import accelerate
|
5 |
|
6 |
models=[
|
7 |
+
"ItsJayQz/Marvel_WhatIf_Diffusion",
|
8 |
+
"DGSpitzer/Cyberpunk-Anime-Diffusion",
|
9 |
+
"DGSpitzer/Guan-Yu-Diffusion",
|
10 |
+
"wavymulder/portraitplus",
|
11 |
+
"nitrosocke/classic-anim-diffusion",
|
12 |
+
"22h/vintedois-diffusion-v0-1",
|
13 |
+
"dreamlike-art/dreamlike-diffusion-1.0",
|
14 |
+
"stabilityai/stable-diffusion-2-1"
|
15 |
]
|
16 |
|
17 |
+
for i in range(len(models)):
|
18 |
+
gr.Interface.load(f"models/{models[i]}",live=True,preprocess=True)
|
19 |
+
|
20 |
def TextToImage(Prompt,model):
|
21 |
model_id = model
|
22 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.get_default_dtype())
|