Ease the run on local

infer.py

@@ -6,7 +6,15 @@ import tempfile
 import numpy as np
 import tqdm
 import ffmpeg
-import spaces
+
+try:
+    import spaces
+except:
+    class spaces():
+        def GPU(*args, **kwargs):
+            def decorator(function):
+                return lambda *dummy_args, **dummy_kwargs: function(*dummy_args, **dummy_kwargs)
+            return decorator
 
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')