Update demo/modules/init_model.py

demo/modules/init_model.py

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