Update app.py

app.py

@@ -16,7 +16,7 @@ from trainer import Trainer
 
 TITLE = '# [Tune-A-Video](https://tuneavideo.github.io/) UI'
 
-ORIGINAL_SPACE_ID = 'Tune-A-Video-library/Tune-A-Video-Training-UI'
+ORIGINAL_SPACE_ID = 'Tune-A-Video-library/Tune-A-Video-Training-UI-poli'
 SPACE_ID = os.getenv('SPACE_ID', ORIGINAL_SPACE_ID)
 GPU_DATA = getoutput('nvidia-smi')
 SHARED_UI_WARNING = f'''## Attention - Training doesn't work in this shared UI. You can duplicate and use it with a paid private T4 GPU.
@@ -61,12 +61,13 @@ pipe = InferencePipeline(HF_TOKEN)
 trainer = Trainer(HF_TOKEN)
 
 with gr.Blocks(css='style.css') as demo:
-    if SPACE_ID != ORIGINAL_SPACE_ID:
+    if SPACE_ID == ORIGINAL_SPACE_ID:
         show_warning(SHARED_UI_WARNING)
-    elif(not "T4" in GPU_DATA):
-        show_warning(INVALID_GPU_WARNING)
     elif not torch.cuda.is_available():
         show_warning(CUDA_NOT_AVAILABLE_WARNING)
+    elif(not "T4" in GPU_DATA):
+        show_warning(INVALID_GPU_WARNING)
+    
 
     gr.Markdown(TITLE)
     with gr.Tabs():