Update app.py

app.py

@@ -1,7 +1,7 @@
 import spaces
 
-import torch
-torch.multiprocessing.set_start_method('spawn', force=True)
+# import torch
+# torch.multiprocessing.set_start_method('spawn', force=True)
 
 import re
 import os
@@ -229,6 +229,9 @@ def parse_args() -> Config:
     return Config(**vars(args))
 
 if __name__ == "__main__":
-    config = Config()
+    import torch.multiprocessing as mp
+    mp.set_start_method('spawn', force=True)  # Corrected start method for CUDA
+    
+    config = Config()  
     demo = create_demo(config.name, config.device, config.offload, config.ckpt_dir)
     demo.launch(share=config.share)