nanushio commited on
Commit
80a6c49
1 Parent(s): 69a2d79

- [MINOR] [SOURCE] [UPDATE] 1. update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -36,9 +36,6 @@ def inference_one_video(input_video):
36
  """
37
  BASIC SETTINGS
38
  """
39
- torch.cuda.current_device()
40
- torch.cuda.empty_cache()
41
- torch.backends.cudnn.benchmark = True
42
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
43
  with open("./cover.yml", "r") as f:
44
  opt = yaml.safe_load(f)
 
36
  """
37
  BASIC SETTINGS
38
  """
 
 
 
39
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
40
  with open("./cover.yml", "r") as f:
41
  opt = yaml.safe_load(f)