Update trainer.py

trainer.py

@@ -20,7 +20,7 @@ from utils import save_model_card
 sys.path.append('Tune-A-Video')
 
 URL_TO_JOIN_MODEL_LIBRARY_ORG = 'https://huggingface.co/organizations/Tune-A-Video-library/share/YjTcaNJmKyeHFpMBioHhzBcTzCYddVErEk'
-ORIGINAL_SPACE_ID = 'Tune-A-Video-library/Tune-A-Video-Training-UI'
+ORIGINAL_SPACE_ID = 'multimodalart/Tune-A-Video-Training-UI-poli'
 SPACE_ID = os.getenv('SPACE_ID', ORIGINAL_SPACE_ID)
 
 class Trainer:
@@ -72,7 +72,7 @@ class Trainer:
         remove_gpu_after_training: bool,
         input_token: str,
     ) -> str:
-        if SPACE_ID != ORIGINAL_SPACE_ID:
+        if SPACE_ID == ORIGINAL_SPACE_ID:
             raise gr.Error('This Space does not work on this Shared UI. Duplicate the Space and attribute a GPU')
         if not torch.cuda.is_available():
             raise gr.Error('CUDA is not available.')