Rooni commited on
Commit
5847e71
1 Parent(s): 71036e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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 LoRA", "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,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 LoRA':
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,