AlStable commited on
Commit
0d7c941
1 Parent(s): 47c4278

Update app.py

Browse files

adding more models

Files changed (1) hide show
  1. app.py +18 -2
app.py CHANGED
@@ -6,10 +6,26 @@ API_KEY=os.environ.get('HUGGING_FACE_HUB_TOKEN', None)
6
  article = """---
7
  This space was created using [SD Space Creator](https://huggingface.co/spaces/anzorq/sd-space-creator)."""
8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  gr.Interface.load(
10
- name="models/stabilityai/stable-diffusion-2-1",
11
  title="""Stable Diffusion 2 1 sandbox""",
12
  description="""Demo for <a href="https://huggingface.co/stabilityai/stable-diffusion-2-1">Stable Diffusion 2 1</a> Stable Diffusion model.""",
13
  article=article,
14
  api_key=API_KEY,
15
- ).queue(concurrency_count=20).launch()
 
6
  article = """---
7
  This space was created using [SD Space Creator](https://huggingface.co/spaces/anzorq/sd-space-creator)."""
8
 
9
+ class Model:
10
+ def __init__(self, name, path="", prefix=""):
11
+ self.name = name
12
+ self.path = path
13
+ self.prefix = prefix
14
+ self.pipe_t2i = None
15
+ self.pipe_i2i = None
16
+
17
+ models = [
18
+ Model("Marvel","ItsJayQz/Marvel_WhatIf_Diffusion", "Marvel")
19
+ Model("Cyberpunk Anime Diffusion", "DGSpitzer/Cyberpunk-Anime-Diffusion", "dgs illustration style "),
20
+ Model("Guan Yu Diffusion", "DGSpitzer/Guan-Yu-Diffusion", "Guan-Yu")
21
+ ]
22
+
23
+ custom_model = "models/stabilityai/stable-diffusion-2-1"
24
+
25
  gr.Interface.load(
26
+ name= custom_model,
27
  title="""Stable Diffusion 2 1 sandbox""",
28
  description="""Demo for <a href="https://huggingface.co/stabilityai/stable-diffusion-2-1">Stable Diffusion 2 1</a> Stable Diffusion model.""",
29
  article=article,
30
  api_key=API_KEY,
31
+ ).queue(concurrency_count=20).launch()