rahul7star commited on
Commit
9e228c1
·
verified ·
1 Parent(s): 2ef8054

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ REPO_ID = "easygoing0114/AI_upscalers"
12
  MODEL_DIR = "/tmp/upscalers"
13
  os.makedirs(MODEL_DIR, exist_ok=True)
14
 
15
- DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
16
 
17
  # -----------------------------
18
  # LOAD MODEL LIST
 
12
  MODEL_DIR = "/tmp/upscalers"
13
  os.makedirs(MODEL_DIR, exist_ok=True)
14
 
15
+ DEVICE = "cpu"
16
 
17
  # -----------------------------
18
  # LOAD MODEL LIST