Update app.py

app.py

@@ -54,7 +54,7 @@ pipe = InferencePipeline(HF_TOKEN)
 trainer = Trainer(HF_TOKEN)
 
 with gr.Blocks(css='style.css') as demo:
-    if os.getenv('IS_SHARED_UI'):
+    if SPACE_ID != ORIGINAL_SPACE_ID:
         show_warning(SHARED_UI_WARNING)
     elif not torch.cuda.is_available():
         show_warning(CUDA_NOT_AVAILABLE_WARNING)