Update main.py

main.py

@@ -300,7 +300,7 @@ def nsfw_check(item, attempts=1):
             return nsfw_check(item, attempts+1)
         
         scores = {item['label']: item['score'] for item in json_response}
-        error_msg = scores.get('nsfw', 0) > scores.get('normal', 0)
+        error_msg = scores.get('nsfw', 0) > .35
         return error_msg
     except json.JSONDecodeError as e:
         print(f'JSON Decoding Error: {e}')
@@ -316,7 +316,6 @@ def nsfw_check(item, attempts=1):
 async def inference(item: Item):
     print("Start API Inference")
     activeModels = InferenceClient().list_deployed_models()
-    print(activeModels['text-to-image'])
     base64_img = ""
     model = item.modelID
     print(f'Start Model {model}')