Update demo/modules/init_model.py

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
 
-read = os.popen("ls /data/ProTrek_650M_UniRef50")
+read = os.popen("ls /data")
 print(read.read())
 def load_model():
     model_config = {