Update app.py

app.py

@@ -7,9 +7,9 @@ import re
 from gradio_client import Client
 
 is_shared_ui = True if "fffiloni/consistent-character" in os.environ['SPACE_ID'] else False
-def safety_check(user_prompt):
+def safety_check(user_prompt, token):
     
-    client = Client("fffiloni/safety-checker-bot", hf_token=hf_token)
+    client = Client("fffiloni/safety-checker-bot", hf_token=token)
     response = client.predict(
             source_space="consistent-character space",
             user_prompt=user_prompt,
@@ -35,7 +35,7 @@ def predict(request: gr.Request, *args, progress=gr.Progress(track_tqdm=True)):
         if is_shared_ui:
             hf_token = os.environ.get("HF_TOKEN")
 
-            is_safe = safety_check(args[0])
+            is_safe = safety_check(args[0], hf_token)
             print(is_safe)
     
             match = re.search(r'\bYes\b', is_safe)