Update app.py

app.py

@@ -5,6 +5,7 @@ import gdown
 import os
 import setup_environment
 import zipfile
+import sys
 from setup_environment import initialize_environment
 
 zero = torch.Tensor([0]).cuda()