Update app.py

app.py

@@ -73,15 +73,15 @@ examples = [
 print(f"### Cleaning cached examples ...")
 os.system(f"rm -rf gradio_cached_examples/")
 
-device = torch.device('cuda:0')
+device = "cuda"
 
 class MagicTimeController:
     def __init__(self):
         # config dirs
         self.basedir                = os.getcwd()
-        self.stable_diffusion_dir   = os.path.join(self.basedir, "ckpts", "Base_Model")
-        self.motion_module_dir      = os.path.join(self.basedir, "ckpts", "Base_Model", "motion_module")
-        self.personalized_model_dir = os.path.join(self.basedir, "ckpts", "DreamBooth")
+        self.stable_diffusion_dir   = os.path.join(self.basedir, model_path, "Base_Model")
+        self.motion_module_dir      = os.path.join(self.basedir, model_path, "Base_Model", "motion_module")
+        self.personalized_model_dir = os.path.join(self.basedir, model_path, "DreamBooth")
         self.savedir                = os.path.join(self.basedir, "outputs")
         os.makedirs(self.savedir, exist_ok=True)