Update level_classifier_tool_2.py

level_classifier_tool_2.py

@@ -21,10 +21,8 @@ class HFEmbeddingBackend:
     MODEL: Any = field(init=False, repr=False)
 
     def __post_init__(self):
-        
         os.environ.setdefault("SPACES_ZERO_DISABLED", "1")
-
-       try:
+        try:
             torch.backends.cuda.sdp_kernel(enable_math=True, enable_flash=False, enable_mem_efficient=False)
         except Exception:
             pass
@@ -65,7 +63,7 @@ class HFEmbeddingBackend:
                 all_out.append(pooled.cpu())
 
         embs = torch.cat(all_out, dim=0) if all_out else torch.empty((0, self.MODEL.config.hidden_size))  # type: ignore
-        return embs, texts_list----------------- Utilities ---------------------------
+        return embs, texts_list
 
 def _normalize_whitespace(s: str) -> str:
     return " ".join(s.strip().split())