1 files changed, 8 insertions, 14 deletions

diff --git a/g4f/Provider/airforce/AirforceImage.py b/g4f/Provider/airforce/AirforceImage.py
index 62d42b4f..b74bc364 100644
--- a/g4f/Provider/airforce/AirforceImage.py
+++ b/g4f/Provider/airforce/AirforceImage.py
@@ -3,6 +3,7 @@ from __future__ import annotations
 from aiohttp import ClientSession
 from urllib.parse import urlencode
 import random
+import requests
 
 from ...typing import AsyncResult, Messages
 from ...image import ImageResponse
@@ -16,20 +17,13 @@ class AirforceImage(AsyncGeneratorProvider, ProviderModelMixin):
     #working = True
     
     default_model = 'flux'
-    image_models = [
-        'flux',
-        'flux-realism', 
-        'flux-anime',
-        'flux-3d',
-        'flux-disney',
-        'flux-pixel',
-        'flux-4o',
-        'any-dark',
-        'stable-diffusion-xl-base',
-        'stable-diffusion-xl-lightning',
-        'Flux-1.1-Pro',
-    ]
-    models = [*image_models]
+    
+    response = requests.get('https://api.airforce/imagine/models')
+    data = response.json()
+
+    image_models = data
+
+    models = [*image_models, "stable-diffusion-xl-base", "stable-diffusion-xl-lightning", "Flux-1.1-Pro"]
     
     model_aliases = {
         "sdxl": "stable-diffusion-xl-base",