NeelakshSaxena commited on
Commit
b9d0270
Β·
1 Parent(s): 38a1317

Lazy-load TMOS classifier for tiny mode

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -18,8 +18,6 @@ SCRIPTS_DIR = ROOT_DIR / "scripts"
18
  if str(SCRIPTS_DIR) not in sys.path:
19
  sys.path.insert(0, str(SCRIPTS_DIR))
20
 
21
- from tmos_classifier import TMOSClassifier
22
-
23
  load_dotenv()
24
  HF_TOKEN = os.getenv("HF_TOKEN")
25
 
@@ -151,6 +149,7 @@ def load_model_and_processor():
151
  return model, processor, inference_device
152
 
153
  from peft import PeftModel
 
154
 
155
  adapter_source = None
156
  local_adapter_file = next(
 
18
  if str(SCRIPTS_DIR) not in sys.path:
19
  sys.path.insert(0, str(SCRIPTS_DIR))
20
 
 
 
21
  load_dotenv()
22
  HF_TOKEN = os.getenv("HF_TOKEN")
23
 
 
149
  return model, processor, inference_device
150
 
151
  from peft import PeftModel
152
+ from tmos_classifier import TMOSClassifier
153
 
154
  adapter_source = None
155
  local_adapter_file = next(