sdart commited on
Commit
3206fc4
1 Parent(s): 5126a09

switch to gpu

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -7,7 +7,7 @@ import torch
7
  import PIL.Image
8
  import numpy as np
9
 
10
- device = "cpu"
11
 
12
  class Model:
13
  def __init__(self):
 
7
  import PIL.Image
8
  import numpy as np
9
 
10
+ device = "cuda"
11
 
12
  class Model:
13
  def __init__(self):