Update app.py

app.py

@@ -3,16 +3,8 @@ import os
 import OpenGL.GL as gl
 os.environ["PYOPENGL_PLATFORM"] = "egl"
 sys.argv = ['VQ-Trans/GPT_eval_multi.py']
-os.umask(0)
-os.makedirs('output', exist_ok=True)
 os.chdir('VQ-Trans')
-os.umask(0)
-# os.system('gdown --fuzzy https://drive.google.com/file/d/1chKRmW2vNX5Lz6dPO45WY1qVNUkoJJ5D/view')
-# os.system('gdown --fuzzy https://drive.google.com/file/d/1cgUVeGwPU7PzTDx20CAijt18fNhtnOPq/view')
-# os.system('gdown --fuzzy https://drive.google.com/file/d/1o7RTDQcToJjTm9_mNWTyzvZvjTWpZfug/view -O checkpoints/')
-# os.system('gdown --fuzzy https://drive.google.com/file/d/1tX79xk0fflp07EZ660Xz1RAFE33iEyJR/view -O checkpoints/')
-os.system('unzip home/user/app/VQ-Trans/checkpoints/t2m.zip')
-os.system('unzip home/user/app/VQ-Trans/checkpoints/kit.zip')
+
 sys.path.append('/home/user/app/VQ-Trans')
 
 import options.option_transformer as option_trans
@@ -100,8 +92,8 @@ ckpt = torch.load(args.resume_trans, map_location='cpu')
 trans_encoder.load_state_dict(ckpt['trans'], strict=True)
 trans_encoder.eval()
 
-mean = torch.from_numpy(np.load('./checkpoints/t2m/VQVAEV3_CB1024_CMT_H1024_NRES3/meta/mean.npy'))
-std = torch.from_numpy(np.load('./checkpoints/t2m/VQVAEV3_CB1024_CMT_H1024_NRES3/meta/std.npy'))
+mean = torch.from_numpy(np.load(f'{model_path}/meta/mean.npy'))
+std = torch.from_numpy(np.load(f'{model_path}/meta/std.npy'))
 
 if is_cuda:
     net.cuda()