estellea commited on
Commit
f4ff0d3
1 Parent(s): bdfdf08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -8,7 +8,6 @@ from tempfile import NamedTemporaryFile
8
  from pathlib import Path
9
 
10
  Path("tmp").mkdir(exist_ok=True)
11
-
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
  print(f"Device is {device}")
14
  torch_type = torch.float16 if device == "cuda" else torch.float32
 
8
  from pathlib import Path
9
 
10
  Path("tmp").mkdir(exist_ok=True)
 
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
  print(f"Device is {device}")
13
  torch_type = torch.float16 if device == "cuda" else torch.float32