codelion commited on
Commit
aa4ca59
·
verified ·
1 Parent(s): de08a40

Use float32 ONNX (exact PyTorch match, faster CPU inference)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from adaptive_classifier import AdaptiveClassifier
10
  # Load model once at startup
11
  print("Loading model...")
12
  classifier = AdaptiveClassifier.from_pretrained(
13
- "adaptive-classifier/ai-detector", use_onnx=False
14
  )
15
  print("Model loaded!")
16
 
 
10
  # Load model once at startup
11
  print("Loading model...")
12
  classifier = AdaptiveClassifier.from_pretrained(
13
+ "adaptive-classifier/ai-detector", use_onnx=True, prefer_quantized=False
14
  )
15
  print("Model loaded!")
16