Update app.py

app.py

@@ -62,7 +62,7 @@ def extract_info(json_data):
                         
                         # Then append all image URLs to the list
                         for image in model_version["images"]:
-                            image_id = image["url"].split("/")[-1]
+                            image_id = image["url"].split("/")[-1].split(".")[0]
                             prompt, negative_prompt = get_prompts_from_image(image_id)
                             if image["nsfwLevel"] > 2:
                                 pass #ugly before checking the actual logic