Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
|
|
11 |
API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
|
12 |
headers = {"Authorization": f"Bearer {API_TOKEN}"}
|
13 |
|
14 |
-
models_list = ["AbsoluteReality 1.8.1", "DALL-E 3 XL", "Playground 2", "Openjourney 4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit 2.5", "Realistic Vision 5.1", "Incursios 1.6", "Anime Detailer XL
|
15 |
|
16 |
def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, seed=None, sampler="DPM++ 2M Karras"):
|
17 |
language = detect(prompt)
|
@@ -40,7 +40,7 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, seed=None, sa
|
|
40 |
API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/realistic-vision-v51"
|
41 |
if model == 'Incursios 1.6':
|
42 |
API_URL = "https://api-inference.huggingface.co/models/digiplay/incursiosMemeDiffusion_v1.6"
|
43 |
-
if model == 'Anime Detailer XL
|
44 |
API_URL = "https://api-inference.huggingface.co/models/Linaqruf/anime-detailer-xl-lora"
|
45 |
if model == 'epiCRealism':
|
46 |
API_URL = "https://api-inference.huggingface.co/models/emilianJR/epiCRealism"
|
@@ -54,6 +54,8 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, seed=None, sa
|
|
54 |
API_URL = "https://api-inference.huggingface.co/models/PixArt-alpha/PixArt-XL-2-1024-MS"
|
55 |
if model == 'Waifu 1.4':
|
56 |
API_URL = "https://api-inference.huggingface.co/models/gisohi6975/nsfw-waifu-diffusion"
|
|
|
|
|
57 |
|
58 |
payload = {
|
59 |
"inputs": prompt,
|
|
|
11 |
API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
|
12 |
headers = {"Authorization": f"Bearer {API_TOKEN}"}
|
13 |
|
14 |
+
models_list = ["AbsoluteReality 1.8.1", "DALL-E 3 XL", "Playground 2", "Openjourney 4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit 2.5", "Realistic Vision 5.1", "Incursios 1.6", "Anime Detailer XL", "Vector Art XL", "epiCRealism", "PixelArt XL", "NewReality XL", "Anything 5.0", "PixArt XL 2.0", "Waifu 1.4"]
|
15 |
|
16 |
def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, seed=None, sampler="DPM++ 2M Karras"):
|
17 |
language = detect(prompt)
|
|
|
40 |
API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/realistic-vision-v51"
|
41 |
if model == 'Incursios 1.6':
|
42 |
API_URL = "https://api-inference.huggingface.co/models/digiplay/incursiosMemeDiffusion_v1.6"
|
43 |
+
if model == 'Anime Detailer XL':
|
44 |
API_URL = "https://api-inference.huggingface.co/models/Linaqruf/anime-detailer-xl-lora"
|
45 |
if model == 'epiCRealism':
|
46 |
API_URL = "https://api-inference.huggingface.co/models/emilianJR/epiCRealism"
|
|
|
54 |
API_URL = "https://api-inference.huggingface.co/models/PixArt-alpha/PixArt-XL-2-1024-MS"
|
55 |
if model == 'Waifu 1.4':
|
56 |
API_URL = "https://api-inference.huggingface.co/models/gisohi6975/nsfw-waifu-diffusion"
|
57 |
+
if model == 'Vector Art XL':
|
58 |
+
API_URL = "https://api-inference.huggingface.co/models/DoctorDiffusion/doctor-diffusion-s-controllable-vector-art-xl-lora"
|
59 |
|
60 |
payload = {
|
61 |
"inputs": prompt,
|