vsimmer commited on
Commit
6830d5b
·
verified ·
1 Parent(s): 91f4f7c

Update handler.py

Browse files

version -07 includes Con sort of dog_faces by Trent

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-04.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-07.bin")
12
  self.model.load_state_dict(torch.load(weights_path, map_location="cpu"))
13
  self.model.eval()
14