Rooni commited on
Commit
3864ee5
1 Parent(s): 5847e71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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", "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)
@@ -52,10 +52,10 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, seed=None, sa
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
- 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,
 
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", "Disney Cartoon"]
15
 
16
  def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, seed=None, sampler="DPM++ 2M Karras"):
17
  language = detect(prompt)
 
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
  if model == 'Vector Art XL':
56
  API_URL = "https://api-inference.huggingface.co/models/DoctorDiffusion/doctor-diffusion-s-controllable-vector-art-xl-lora"
57
+ if model == 'Disney Cartoon':
58
+ API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/disney-pixal-cartoon"
59
 
60
  payload = {
61
  "inputs": prompt,