Update app.py

app.py

@@ -11,23 +11,25 @@ import json
 from random import randint
 
 # API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
-API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
+API_URL = "black-forest-labs/FLUX.1-schnell"
 
 mod_list = {
     "FLUX.1 Schnell": API_URL,
-    "FLUX.1 Schnell | Face Realism": "https://api-inference.huggingface.co/models/prithivMLmods/Canopus-LoRA-Flux-FaceRealism",
-    "FLUX.1 Schnell | Midjourney": "https://api-inference.huggingface.co/models/Jovie/Midjourney_Schnell",
-    "FLUX.1 Dev": "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev",
-    "FLUX.1 Schnell | realism": "https://api-inference.huggingface.co/models/hugovntr/flux-schnell-realism",
-    "FLUX.1 Schnell | MJ v6": "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Midjourney-Mix2-LoRA",
-    "FLUX.1 Dev | Flux Realism": "https://api-inference.huggingface.co/models/XLabs-AI/flux-RealismLora",
-    "FLUX.1 Redux": "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-Redux-dev",
-    "FLUX.1 Dev | UltraRealsim 2.0": "https://api-inference.huggingface.co/models/prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0",
-    "Lumina Image 2.0": "https://api-inference.huggingface.co/models/Alpha-VLLM/Lumina-Image-2.0",
-    "FLUX.1 Dev | Vector Journey": "https://api-inference.huggingface.co/models/Shakker-Labs/FLUX.1-dev-LoRA-Vector-Journey",
-    "FLUX.1 Dev | Deep Blue": "https://api-inference.huggingface.co/models/fffiloni/deep-blue-v2",
+    "FLUX.1 Schnell | Face Realism": "prithivMLmods/Canopus-LoRA-Flux-FaceRealism",
+    "FLUX.1 Schnell | Midjourney": "Jovie/Midjourney_Schnell",
+    "FLUX.1 Dev": "black-forest-labs/FLUX.1-dev",
+    "FLUX.1 Schnell | realism": "hugovntr/flux-schnell-realism",
+    "FLUX.1 Schnell | MJ v6": "strangerzonehf/Flux-Midjourney-Mix2-LoRA",
+    "FLUX.1 Dev | Flux Realism": "XLabs-AI/flux-RealismLora",
+    "FLUX.1 Redux": "black-forest-labs/FLUX.1-Redux-dev",
+    "FLUX.1 Dev | UltraRealsim 2.0": "prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0",
+    "Lumina Image 2.0": "Alpha-VLLM/Lumina-Image-2.0",
+    "FLUX.1 Dev | Vector Journey": "Shakker-Labs/FLUX.1-dev-LoRA-Vector-Journey",
+    "FLUX.1 Dev | Deep Blue": "fffiloni/deep-blue-v2",
     "Realism Engine v1.0": "digiplay/RealismEngine_v1",
-    "Absolute Reality v1.8.1": "digiplay/AbsoluteReality_v1.8.1"
+    "Absolute Reality v1.8.1": "digiplay/AbsoluteReality_v1.8.1",
+    "FLUX.1 Dev | Midjourney Anime": "brushpenbob/flux-midjourney-anime",
+    "FLUX.1 Dev | ": "brushpenbob/flux-pencil-v2"
 }
 API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
@@ -126,6 +128,8 @@ def query(prompt, is_realistic, num_inference_steps, width, height, mod = None):
     for mod_name, mod_link in mod_list.items():
         if (mod == mod_name):
             model = mod_link
+
+    model = "https://api-inference.huggingface.co/models/" + model;
     
     response = requests.post(model, headers=headers, json=payload, timeout=timeout)
     if response.status_code != 200: