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

Switch back to PyTorch inference - ONNX quantization flips borderline predictions

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=True
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=False
14
  )
15
  print("Model loaded!")
16