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 v1.8.1", "DALL-E 3 XL", "Playground v2", "Openjourney v4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit v2.5", "Realistic Vision 5.1", "Incursios v1.6", "Anime Detailer XL LoRA", "epiCRealism", "PixelArt XL", "NewReality XL"]
+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", "Midjourney 4.0"]
 
 def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
     language = detect(prompt)
@@ -22,21 +22,21 @@ def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
     
     if model == 'DALL-E 3 XL':
         API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
-    if model == 'Playground v2':
+    if model == 'Playground 2':
         API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
-    if model == 'Openjourney v4':
+    if model == 'Openjourney 4':
         API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney-v4"
-    if model == 'AbsoluteReality v1.8.1':
+    if model == 'AbsoluteReality 1.8.1':
         API_URL = "https://api-inference.huggingface.co/models/digiplay/AbsoluteReality_v1.8.1"
     if model == 'Lyriel 1.6':
         API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/lyrielv16"
     if model == 'Animagine XL 2.0':
         API_URL = "https://api-inference.huggingface.co/models/Linaqruf/animagine-xl-2.0"
-    if model == 'Counterfeit v2.5':
+    if model == 'Counterfeit 2.5':
         API_URL = "https://api-inference.huggingface.co/models/gsdf/Counterfeit-V2.5"
     if model == 'Realistic Vision 5.1':
         API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/realistic-vision-v51"
-    if model == 'Incursios v1.6':
+    if model == 'Incursios 1.6':
         API_URL = "https://api-inference.huggingface.co/models/digiplay/incursiosMemeDiffusion_v1.6"
     if model == 'Anime Detailer XL LoRA':
         API_URL = "https://api-inference.huggingface.co/models/Linaqruf/anime-detailer-xl-lora"
@@ -46,6 +46,8 @@ def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
         API_URL = "https://api-inference.huggingface.co/models/nerijs/pixel-art-xl"
     if model == 'NewReality XL':
         API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/newrealityxl-global-nsfw"
+    if model == 'Midjourney 4.0':
+        API_URL = "https://api-inference.huggingface.co/models/openskyml/midjourney-v4-xl"
     
     payload = {
         "inputs": prompt,