Changed ckpt path

app.py

@@ -225,8 +225,8 @@ class Drag:
         self.device = device
 
         ckpts_dir = "ckpts/"
-        svd_ckpt = "ckpts/stable-video-diffusion-img2vid-xt-1-1"
-        mofa_ckpt = "ckpts/controlnet"
+        svd_ckpt = "./ckpts/stable-video-diffusion-img2vid-xt-1-1"
+        mofa_ckpt = "./ckpts/controlnet"
 
         self.device = 'cuda'
         self.weight_dtype = torch.float16
@@ -647,26 +647,6 @@ class Drag:
 
 
 with gr.Blocks() as demo:
-    import os
-    # import nvidia.cublas.lib
-    # import nvidia.cudnn.lib
-    # # print(os.path.dirname(nvidia.cublas.lib.__file__))
-    # print(os.path.dirname(nvidia.cudnn.lib.__file__))
-
-    # /usr/local/lib/python3.10/site-packages/nvidia/cudnn/lib
-
-    import os
-
-    folder_path = "/usr/local/lib/python3.10/site-packages/nvidia/cusolver/lib"
-    if os.path.exists(folder_path) and os.path.isdir(folder_path):
-        print(f"Contents of the folder '{folder_path}':")
-        for item in os.listdir(folder_path):
-            print(item)
-    else:
-        print(f"The folder '{folder_path}' does not exist or is not a directory.")
-
-    print(os.environ["LD_LIBRARY_PATH"])
-
     gr.Markdown("""<h1 align="center">MOFA-Video</h1><br>""")
 
     gr.Markdown("""Official Gradio Demo for <a href='https://myniuuu.github.io/MOFA_Video'><b>MOFA-Video: Controllable Image Animation via Generative Motion Field Adaptions in Frozen Image-to-Video Diffusion Model</b></a>.<br>""")