DOFOFFICIAL commited on
Commit
2671a55
1 Parent(s): b812c98

Update usage/interfere_cpu.py

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