Update app.py
Browse files
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,
|