Update app.py

app.py

@@ -11,7 +11,7 @@ API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
 API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
-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"]
+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"]
 
 def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, seed=None, sampler="DPM++ 2M Karras"):
     language = detect(prompt)
@@ -56,6 +56,10 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, seed=None, sa
         API_URL = "https://api-inference.huggingface.co/models/DoctorDiffusion/doctor-diffusion-s-controllable-vector-art-xl-lora"
     if model == 'Disney Cartoon':
         API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/disney-pixal-cartoon"
+    if model == 'CleanLinearMix':
+        API_URL = "https://api-inference.huggingface.co/models/digiplay/CleanLinearMix_nsfw"
+    if model == 'Waifu 1.4':
+        API_URL = "https://api-inference.huggingface.co/models/gisohi6975/nsfw-waifu-diffusion"
     
     payload = {
         "inputs": prompt,