Update app.py

app.py

@@ -4,25 +4,10 @@ import requests
 import time
 import os
 import re
-
-hf_token = os.environ.get("HF_TOKEN")
 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):
-    
-    client = Client("fffiloni/safety-checker-bot", hf_token=hf_token)
-    response = client.predict(
-    		source_space="consistent-character space",
-    		user_prompt=user_prompt,
-    		api_name="/infer"
-    )
-    print(response)
-    
-    return response
-
 from utils.gradio_helpers import parse_outputs, process_outputs
 
 names = ['prompt', 'negative_prompt', 'subject', 'number_of_outputs', 'number_of_images_per_pose', 'randomise_poses', 'output_format', 'output_quality', 'seed']
@@ -37,6 +22,20 @@ def predict(request: gr.Request, *args, progress=gr.Progress(track_tqdm=True)):
     
     try:
         if is_shared_ui:
+            hf_token = os.environ.get("HF_TOKEN")
+            
+            def safety_check(user_prompt):
+    
+                client = Client("fffiloni/safety-checker-bot", hf_token=hf_token)
+                response = client.predict(
+                		source_space="consistent-character space",
+                		user_prompt=user_prompt,
+                		api_name="/infer"
+                )
+                print(response)
+                
+                return response
+            
             is_safe = safety_check(args[0])
             print(is_safe)