Update app.py

app.py

@@ -32,7 +32,7 @@ def check_nsfw(json_data, profile):
     for model_version in json_data["modelVersions"]:
         for image in model_version["images"]:
             print(image)
-            if image["nsfw"] not in ["None", "Soft"]:
+            if image["nsfw"] > 2:
                 return False
     return True
 
@@ -47,8 +47,7 @@ def extract_info(json_data):
                         
                         # Then append all image URLs to the list
                         for image in model_version["images"]:
-                            print(image)
-                            if image["nsfw"] != "None":
+                            if image["nsfwLevel"] > 2:
                                 pass #ugly before checking the actual logic
                             else:
                                 urls_to_download.append({