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())
|