disable safety checker on non shared ui

app.py

@@ -36,19 +36,19 @@ def predict(request: gr.Request, *args, progress=gr.Progress(track_tqdm=True)):
         raise gr.Error(f"You forgot to provide a prompt.")
     
     try:
+        if is_shared_ui:
+            is_safe = safety_check(args[0])
+            print(is_safe)
         
-        is_safe = safety_check(args[0])
-        print(is_safe)
-    
-        match = re.search(r'\bYes\b', is_safe)
-    
-        if match:
-            status = 'Yes'
-        else:
-            status = None
-    
-        if status == "Yes" :
-            raise gr.Error("Do not ask for such things.")
+            match = re.search(r'\bYes\b', is_safe)
+        
+            if match:
+                status = 'Yes'
+            else:
+                status = None
+        
+            if status == "Yes" :
+                raise gr.Error("Do not ask for such things.")
         else:
     
             headers = {'Content-Type': 'application/json'}