DOFOFFICIAL commited on
Commit
75b5dc2
·
1 Parent(s): 2671a55

Update usage/interfere.py

Browse files
Files changed (1) hide show
  1. usage/interfere.py +5 -1
usage/interfere.py CHANGED
@@ -81,7 +81,11 @@ def predictmain(model, filepath):
81
 
82
  if __name__ == '__main__':
83
  # local usage
84
- model = modelload("animeGender-dvgg-0.7.pth")
 
 
 
 
85
 
86
  # use your picture to interfere
87
  cmpgraph_64x64("path.png", "path(1).png")
 
81
 
82
  if __name__ == '__main__':
83
  # local usage
84
+ model = modelload("pytorch_model.bin")
85
+
86
+ # or uses safetensors
87
+ # model = torch.load("pytorch_model.safetensors")
88
+ # model = model.to(torch.device('cuda'))
89
 
90
  # use your picture to interfere
91
  cmpgraph_64x64("path.png", "path(1).png")