yizhangliu commited on
Commit
ba43f2b
·
1 Parent(s): 37735a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -12,9 +12,12 @@ def getTextTrans(text, source='zh', target='en'):
12
  except Exception as e:
13
  return text
14
 
15
- model_ids = {"models/runwayml/stable-diffusion-v1-5":"stable-diffusion-v1-5",
16
- "models/stabilityai/stable-diffusion-2":"stable-diffusion-2",
17
- "models/prompthero/openjourney":"openjourney",
 
 
 
18
  }
19
  tab_actions = []
20
  tab_titles = []
 
12
  except Exception as e:
13
  return text
14
 
15
+ model_ids = {
16
+ "models/CompVis/stable-diffusion-v1-4":"sd-v1-4",
17
+ "models/runwayml/stable-diffusion-v1-5":"sd-v1-5",
18
+ "models/stabilityai/stable-diffusion-2":"sd-v2-0",
19
+ "models/prompthero/openjourney":"openjourney",
20
+ "models/hakurei/waifu-diffusion":"waifu-diffusion",
21
  }
22
  tab_actions = []
23
  tab_titles = []