File size: 1,090 Bytes
4de66a5
 
 
 
 
 
e88fa1a
979d507
e88fa1a
 
 
 
 
 
 
 
0d7c941
 
979d507
 
7163d04
979d507
7163d04
979d507
 
 
963cea5
7163d04
a9effc6
979d507
4de66a5
979d507
7163d04
 
 
acda1ca
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import os
import gradio as gr

API_KEY=os.environ.get('HUGGING_FACE_HUB_TOKEN', None)
article = """---
This space was created using [SD Space Creator](https://huggingface.co/spaces/anzorq/sd-space-creator)."""

models = [
"models/ItsJayQz/Marvel_WhatIf_Diffusion",
"models/DGSpitzer/Cyberpunk-Anime-Diffusion",
"models/DGSpitzer/Guan-Yu-Diffusion",
"models/wavymulder/portraitplus",
"models/nitrosocke/classic-anim-diffusion",
"models/22h/vintedois-diffusion-v0-1",
"models/dreamlike-art/dreamlike-diffusion-1.0",
"models/stabilityai/stable-diffusion-2-1"
]

custom_model = "models/dreamlike-art/dreamlike-diffusion-1.0"

def selectModel(message, models):
    message = message.lower()
    return models

sandbox = gr.Interface.load(
    fn= selectModel,
    title="AlStable sandbox text to img",
    inputs = ["text", gr.Dropdown(models)],
    outputs = "image", 
    description="""Demo for <a href="https://huggingface.co/stabilityai/stable-diffusion-2-1">AlStable</a> Stable Diffusion model.""",
    article=article,
    api_key=API_KEY
)

sandbox.queue(concurrency_count=20).launch()