Update app.py

app.py

@@ -9,15 +9,14 @@ import time
 import demo_util
 import os
 import spaces
+from huggingface_hub import hf_hub_download
 
 model2ckpt = {
     "TiTok-L-32": ("tokenizer_titok_l32.bin", "generator_titok_l32.bin"),
 }
 
-if not os.path.exists("tokenizer_titok_l32.bin"):
-    os.system("gdown 1I_m2Vm4JgQsa7bZVORj-nVhP8fgQLngd")
-if not os.path.exists("generator_titok_l32.bin"):
-    os.system("gdown 1IgqZ_vwGIj2ZWOPuCzilxeQ2UrMVY93l")
+hf_hub_download(repo_id="fun-research/TiTok", filename="tokenizer_titok_l32.bin", local_dir="./")
+hf_hub_download(repo_id="fun-research/TiTok", filename="generator_titok_l32.bin", local_dir="./")
 
 # @spaces.GPU
 def load_model():