Rooni commited on
Commit
bddaca0
1 Parent(s): daa9401

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -50,6 +50,8 @@ def query(prompt, model, is_negative=False, steps=100, cfg_scale=7, seed=-1):
50
  API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/newrealityxl-global-nsfw"
51
  if model == 'Anything 5.0':
52
  API_URL = "https://api-inference.huggingface.co/models/hogiahien/anything-v5-edited"
 
 
53
 
54
  payload = {
55
  "inputs": prompt,
 
50
  API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/newrealityxl-global-nsfw"
51
  if model == 'Anything 5.0':
52
  API_URL = "https://api-inference.huggingface.co/models/hogiahien/anything-v5-edited"
53
+ if model == 'PixArt XL 2.0':
54
+ API_URL = "https://api-inference.huggingface.co/models/PixArt-alpha/PixArt-XL-2-1024-MS"
55
 
56
  payload = {
57
  "inputs": prompt,