Update app.py

app.py

@@ -2,13 +2,21 @@
 # Copyright @ 2023 wdcqc/aieud project.
 # Open-source under license obtainable in project root (LICENSE.md).
 import argparse
+import os
 import urllib.request
-from wfd.webui import start_demo
 
 if __name__ == "__main__":
+ os.makedirs("/home/user/.local/lib/python3.8/site-packages/wfd/mpqapi/")
  # try to fix the CDLL problem
  urllib.request.urlretrieve("https://github.com/wdcqc/WaveFunctionDiffusion/raw/remaster/wfd/mpqapi/libstorm.so", "/home/user/.local/lib/python3.8/site-packages/wfd/mpqapi/libstorm.so")
+ urllib.request.urlretrieve("https://github.com/wdcqc/WaveFunctionDiffusion/raw/remaster/wfd/mpqapi/libstorm.so", "libstorm.so")
 
+ if os.path.exists("/home/user/.local/lib/python3.8/site-packages/wfd/mpqapi/libstorm.so"):
+ print("libstorm EXISTS!!!")
+ else:
+ print("libstorm DOES NOT EXIST!!!")
+ 
+ from wfd.webui import start_demo
  class Arguments:
  colab = False
  link_to_colab = True