Commit
·
75b5dc2
1
Parent(s):
2671a55
Update usage/interfere.py
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
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")
|