manikumargouni commited on
Commit
916a09e
·
verified ·
1 Parent(s): 2b0e686

Upload inference_iab_classifier.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. inference_iab_classifier.py +4 -1
inference_iab_classifier.py CHANGED
@@ -1,6 +1,9 @@
1
  import json
2
 
3
- from iab_classifier import predict_iab_content_classifier
 
 
 
4
 
5
 
6
  def predict(text: str, confidence_threshold: float | None = None):
 
1
  import json
2
 
3
+ try:
4
+ from .iab_classifier import predict_iab_content_classifier # type: ignore
5
+ except ImportError:
6
+ from iab_classifier import predict_iab_content_classifier
7
 
8
 
9
  def predict(text: str, confidence_threshold: float | None = None):