Upload app.py

app.py

@@ -23,14 +23,6 @@ MAX_SEED = np.iinfo(np.int32).max
 TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
 os.makedirs(TMP_DIR, exist_ok=True)
 
-torch.set_grad_enabled(False)
-device = torch.device('cuda')
-os.environ['MASTER_ADDR'] = 'localhost'
-os.environ['MASTER_PORT'] = '12356'
-os.environ["CUDA_VISIBLE_DEVICES"] = "0"
-os.environ["RANK"] = "0"
-os.environ["WORLD_SIZE"] = "1"
-
 
 def start_session(req: gr.Request):
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))