NOT-OMEGA commited on
Commit
f9f9556
·
verified ·
1 Parent(s): d11b204

Update processor_bert.py

Browse files
Files changed (1) hide show
  1. processor_bert.py +1 -1
processor_bert.py CHANGED
@@ -39,7 +39,7 @@ def preload_models():
39
 
40
  _classifier = pipeline(
41
  task="text-classification",
42
- model=MODEL_NAME,
43
  device=-1, # CPU
44
  top_k=1
45
  )
 
39
 
40
  _classifier = pipeline(
41
  task="text-classification",
42
+ model=MODEL_PATH,
43
  device=-1, # CPU
44
  top_k=1
45
  )