Update

demo/modules/init_model.py

@@ -10,7 +10,7 @@ from utils.constants import sequence_level
 from model.ProTrek.protrek_trimodal_model import ProTrekTrimodalModel
 from tqdm import tqdm
 
-
+os.system("ls /data/ProTrek_650M_UniRef50")
 def load_model():
     model_config = {
         "protein_config": glob.glob(f"{config.model_dir}/esm2_*")[0],