Update demo/modules/init_model.py

demo/modules/init_model.py

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