Rooni commited on
Commit
76a5a4f
1 Parent(s): fbd170f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -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")
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", "Disney Cartoon", "CleanLinearMix", "Waifu 1.4"]
15
 
16
  def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1):
17
  if prompt == "" or prompt == None:
@@ -64,8 +64,6 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
64
  API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/disney-pixal-cartoon"
65
  if model == 'CleanLinearMix':
66
  API_URL = "https://api-inference.huggingface.co/models/digiplay/CleanLinearMix_nsfw"
67
- if model == 'Waifu 1.4':
68
- API_URL = "https://api-inference.huggingface.co/models/gisohi6975/nsfw-waifu-diffusion"
69
 
70
  payload = {
71
  "inputs": prompt,
 
11
  API_TOKEN = os.getenv("HF_READ_TOKEN")
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", "Disney Cartoon", "CleanLinearMix"]
15
 
16
  def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1):
17
  if prompt == "" or prompt == None:
 
64
  API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/disney-pixal-cartoon"
65
  if model == 'CleanLinearMix':
66
  API_URL = "https://api-inference.huggingface.co/models/digiplay/CleanLinearMix_nsfw"
 
 
67
 
68
  payload = {
69
  "inputs": prompt,