AlStable commited on
Commit
c1e0059
1 Parent(s): 973ad2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -10
app.py CHANGED
@@ -3,16 +3,14 @@ import torch
3
  import gradio as gr
4
  import accelerate
5
 
6
-
7
  models=[
8
- gr.Interface.load("ItsJayQz/Marvel_WhatIf_Diffusion"),
9
- gr.Interface.load("DGSpitzer/Cyberpunk-Anime-Diffusion"),
10
- gr.Interface.load("DGSpitzer/Guan-Yu-Diffusion"),
11
- gr.Interface.load("wavymulder/portraitplus"),
12
- gr.Interface.load("nitrosocke/classic-anim-diffusion"),
13
- gr.Interface.load("22h/vintedois-diffusion-v0-1"),
14
- gr.Interface.load("dreamlike-art/dreamlike-diffusion-1.0"),
15
- gr.Interface.load("stabilityai/stable-diffusion-2-1")
16
  ]
17
 
18
  def TextToImage(Prompt,model):
@@ -25,7 +23,7 @@ def TextToImage(Prompt,model):
25
 
26
  sandbox = gr.Interface(fn=TextToImage,
27
  inputs=["text", gr.Dropdown(models)],
28
- outputs="image",
29
  title='AlStable Text to Image')
30
 
31
  sandbox.launch()
 
3
  import gradio as gr
4
  import accelerate
5
 
 
6
  models=[
7
+ gr.Interface.load("Marvel","models/ItsJayQz/Marvel_WhatIf_Diffusion", "whatif style"),
8
+ gr.Interface.load("Cyberpunk Anime Diffusion", "models/DGSpitzer/Cyberpunk-Anime-Diffusion", "dgs illustration style"),
9
+ gr.Interface.load("Portrait plus", "models/wavymulder/portraitplus", "portrait+ style"),
10
+ gr.Interface.load("classic Disney", "models/nitrosocke/classic-anim-diffusion", "classic disney style"),
11
+ gr.Interface.load("vintedois", "models/22h/vintedois-diffusion-v0-1", "vintedois style"),
12
+ gr.Interface.load("dreamlike", "models/dreamlike-art/dreamlike-diffusion-1.0","dreamlike style"),
13
+ gr.Interface.load("SD21","models/stabilityai/stable-diffusion-2-1", "sd21 default style")
 
14
  ]
15
 
16
  def TextToImage(Prompt,model):
 
23
 
24
  sandbox = gr.Interface(fn=TextToImage,
25
  inputs=["text", gr.Dropdown(models)],
26
+ outputs="image",
27
  title='AlStable Text to Image')
28
 
29
  sandbox.launch()