Spaces:
Running
Running
Update app.py
Browse files
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', '
|
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]),
|