Update handler.py

handler.py

@@ -114,7 +114,7 @@ class EndpointHandler():
         elapsed = time.perf_counter() - start
         print(f"Model loaded in {elapsed:.2f}s")
         self.tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
-        self.labels = ["High", "Low", "Medium", "UNKNOWN"]  # Update based on your dataset
+        self.labels = ["High", "Latent", "Medium", "None"]  # Update based on your dataset
         print("Compiling model...")
         start = time.perf_counter()
         self.model.compile()