Update inference_manager.py

inference_manager.py

@@ -73,10 +73,9 @@ class AuthHelper:
 
     def check_auth(self, request, token):
         # Extract parameters from the request
-        params = dict(request.query_params)
-        if params.get("_skip_token_passkey", "") == "nsfwaisio_125687":
+        if params.get("_skip_token_passkey", "") == "nsfwaisio_125687" or not request:
             return True
-
+        params = dict(request.query_params)
         # Gather request-specific information
         sip = request.client.host
         shost = request.headers.get("Host", "")
@@ -322,6 +321,7 @@ class ModelManager:
         self.ext_models = {}
         self.model_directory = model_directory
         self.load_models()
+        self.auth_helper = AuthHelper()
         
     #not enabled at the moment
     def load_instant_x(self):
@@ -416,7 +416,7 @@ class ModelManager:
 
     @spaces.GPU(duration=40)
     def generate_with_faceid(self, model_id, request, inference_params, progress=gr.Progress(track_tqdm=True)):
-        auth_helper.check_auth(request, token)
+        self.auth_helper.check_auth(request, token)
         model = self.models.get(model_id)
         if not model:
             raise Exception(f"invalid model_id {model_id}")
@@ -495,7 +495,7 @@ class ModelManager:
                 pipe._guidance_scale = 0.0
 
             return callback_kwargs
-        auth_helper.check_auth(request, token)
+        self.auth_helper.check_auth(request, token)
         model = self.models.get(model_id)
         if not model:
             raise Exception(f"invalid model_id {model_id}")