fix: identification of official demo instance

app.py

@@ -6,7 +6,10 @@ from torch import cuda
 
 from detikzify.webui import build_ui, make_light, BANNER
 
-if is_public:=(getenv("SPACE_ID") == "nllg/DeTikZify") and not cuda.is_available():
+def is_official_demo():
+    return getenv("SPACE_AUTHOR_NAME") == "nllg"
+
+if is_official_demo() and not cuda.is_available():
     center = ".gradio-container {text-align: center}"
     with gr.Blocks(css=center, theme=make_light(gr.themes.Soft()), title="DeTikZify") as demo:
         badge = "https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-xl.svg"
@@ -21,8 +24,7 @@ if is_public:=(getenv("SPACE_ID") == "nllg/DeTikZify") and not cuda.is_available
 else:
     use_big_models = cuda.is_available() and cuda.get_device_properties(0).total_memory > 15835660288
     model = f"detikzify-ds-{'7' if use_big_models else '1.3'}b"
-    algorithm = "sampling" if is_public else "mcts"
-    demo = build_ui(lock=is_public, model=model, algorithm=algorithm, light=True).queue()
+    demo = build_ui(lock=is_official_demo(), model=model, light=True).queue()
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)