Update app.py

app.py

@@ -51,7 +51,7 @@ def get_prompts_from_image(image_id):
 def extract_info(json_data):
     if json_data["type"] == "LORA":
         for model_version in json_data["modelVersions"]:
-            if model_version["baseModel"] in ["SDXL 1.0", "SDXL 0.9", "SD 1.5", "SD 1.4", "SD 2.1", "SD 2.0", "SD 2.0 768", "SD 2.1 768", "Flux.1 D", "Flux.1 S"]:
+            if model_version["baseModel"] in ["SDXL 1.0", "SDXL 0.9", "SD 1.5", "SD 1.4", "SD 2.1", "SD 2.0", "SD 2.0 768", "SD 2.1 768", "SD 3", "Flux.1 D", "Flux.1 S"]:
                 for file in model_version["files"]:
                     print(file)
                     if "primary" in file:
@@ -81,6 +81,7 @@ def extract_info(json_data):
                             "SD 2.0": "stabilityai/stable-diffusion-2-base",
                             "SD 2.1 768": "stabilityai/stable-diffusion-2-1",
                             "SD 2.0 768": "stabilityai/stable-diffusion-2",
+                            "SD 3": "stabilityai/stable-diffusion-3-medium-diffusers",
                             "Flux.1 D": "black-forest-labs/FLUX.1-dev",
                             "Flux.1 S": "black-forest-labs/FLUX.1-schnell"
                         }