reverted NSFW IP blacklisting; doesn't seem to work as expected on HF

app.py

@@ -94,16 +94,10 @@ def parse_checkboxes(checkboxes):
 # ----------------- CORE FUNCTION ----------------- #
 
 CATEGORY_API_MAPPING = {"Top": "tops", "Bottom": "bottoms", "Full-body": "one-pieces"}
-NSFW_ATTEMPTS = {}
 
 
-async def get_tryon_result(model_image, garment_image, category, checkboxes, request: gr.Request):
+async def get_tryon_result(model_image, garment_image, category, checkboxes):
     logger.info("Starting new try-on request...")
-    client_ip = request.client.host
-
-    if NSFW_ATTEMPTS.get(client_ip, 0) >= 3:
-        logger.info(f"Blocking request from blacklisted IP: {client_ip}")
-        return NSFW
 
     # preprocessing: convert to RGB, resize, encode to base64
     model_image, garment_image = map(lambda x: cv2.cvtColor(x, cv2.COLOR_RGB2BGR), [model_image, garment_image])
@@ -149,8 +143,6 @@ async def get_tryon_result(model_image, garment_image, category, checkboxes, req
                 error_msg = f"Prediction failed: {error}"
                 logger.error(error_msg)
                 if "NSFW" in error:
-                    logger.info(f"NSFW attempt IP address: {client_ip}")
-                    NSFW_ATTEMPTS[client_ip] = NSFW_ATTEMPTS.get(client_ip, 0) + 1
                     return NSFW
                 raise Exception(error_msg)
 
@@ -228,4 +220,5 @@ with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Monochrome(radius_size=sizes.radi
 if __name__ == "__main__":
     ip = requests.get("http://ifconfig.me/ip", timeout=1).text.strip()
     logger.info(f"VM IP address: {ip}")
-    demo.launch()
+    demo.queue(default_concurrency_limit=3)
+    demo.launch(share=False)