Update app.py

app.py

@@ -18,7 +18,10 @@ from huggingface_hub import snapshot_download, update_repo_visibility, HfApi
 
 
 is_spaces = True if "SPACE_ID" in os.environ else False
-is_shared_ui = True if "IS_SHARED_UI" in os.environ else False
+if(is_spaces):
+    is_shared_ui = True if "multimodalart/dreambooth-training" in os.environ['SPACE_ID'] else False
+else:
+    is_shared_ui = False
 is_gpu_associated = torch.cuda.is_available()
 
 css = '''