Update app.py

app.py

@@ -65,6 +65,12 @@ FREE_MODEL_TEST_KEY = (
     "sk-bmjbjzleaqfgtqfzmcnsbagxrlohriadnxqrzfocbizaxukw"
 )
 
+FREE_IMAGE_LIST = [
+    "stabilityai/stable-diffusion-3-5-large",
+    "black-forest-labs/FLUX.1-schnell",
+    "stabilityai/stable-diffusion-3-medium"
+]
+
 def test_model_availability(api_key, model_name):
     """
     测试指定的模型是否可用。
@@ -224,14 +230,10 @@ def test_embedding_model_availability(api_key, model_name):
         
 def test_image_model_availability(api_key, model_name):
     """
-    测试指定的向量模型是否可用。
+    测试指定的图像模型是否在 FREE_IMAGE_LIST 中。
+    如果在列表中，返回 True，否则返回 False。
     """
-    headers = {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json"
-    }
-
-    return True
+    return model_name in FREE_IMAGE_LIST
 
 def load_keys():
     """