Arjun24420
/

BERT-FakeOrReal-BinaryClassification

Text Classification

Inference Endpoints

Model card Files Files and versions Community

Arjun24420 commited on Dec 10, 2023

Commit

a4996d6

•

1 Parent(s): 383ea9a

Update README.md

Files changed (1) hide show

README.md +6 -0

README.md CHANGED Viewed

@@ -28,4 +28,10 @@ def predict(text):
     return class_probabilities
 ```

     return class_probabilities
+# Define class labels mapping
+class_mapping = {
+    1: 'Reliable',
+    0: 'Unreliable',
+}
 ```