iimran commited on
Commit
565a2a0
·
verified ·
1 Parent(s): aa14789

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ class ONNXInferencePipeline:
55
  predicted_class = int(np.argmax(probabilities))
56
 
57
  # Map to labels
58
- class_labels = ['Inappropriate Content', 'Not Inappropriate']
59
  return {
60
  'label': class_labels[predicted_class],
61
  'confidence': float(probabilities[0][predicted_class]),
 
55
  predicted_class = int(np.argmax(probabilities))
56
 
57
  # Map to labels
58
+ class_labels = ['Inappropriate Content', 'Appropriate']
59
  return {
60
  'label': class_labels[predicted_class],
61
  'confidence': float(probabilities[0][predicted_class]),