Spaces:

PraneshJs
/

fakevideodetect

Running

PraneshJs commited on Aug 13

Commit

2b0e2be

verified ·

1 Parent(s): 9ce33d8

Update inference_2.py

Files changed (1) hide show

inference_2.py CHANGED Viewed

@@ -12,7 +12,7 @@ from models import image
 # Load ONNX -> PyTorch safely
 # -----------------------------
 onnx_model = onnx.load('checkpoints/efficientnet.onnx')
-pytorch_model = ConvertModel(onnx_model, strict=False)
 torch.manual_seed(42)

 # Load ONNX -> PyTorch safely
 # -----------------------------
 onnx_model = onnx.load('checkpoints/efficientnet.onnx')
+pytorch_model = ConvertModel(onnx_model)
 torch.manual_seed(42)