vsimmer commited on
Commit
59a758f
·
verified ·
1 Parent(s): 8665cd4

Update handler.py

Browse files

move to -04 model which is made with Con's sort

Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -8,7 +8,7 @@ class EndpointHandler():
8
  def __init__(self, path=""):
9
  # Load model and weights
10
  self.model = get_model()
11
- weights_path = os.path.join(path, "doge_223_sd-03.bin")
12
  self.model.load_state_dict(torch.load(weights_path, map_location="cpu"))
13
  self.model.eval()
14
 
 
8
  def __init__(self, path=""):
9
  # Load model and weights
10
  self.model = get_model()
11
+ weights_path = os.path.join(path, "doge_223_sd-04.bin")
12
  self.model.load_state_dict(torch.load(weights_path, map_location="cpu"))
13
  self.model.eval()
14