SreyaDvn commited on
Commit
49d4f01
·
verified ·
1 Parent(s): caec164

Update predictor.py

Browse files
Files changed (1) hide show
  1. predictor.py +6 -0
predictor.py CHANGED
@@ -1,3 +1,9 @@
 
 
 
 
 
 
1
  import os
2
 
3
  # Force Hugging Face to use /tmp as cache
 
1
+ # predictor.py
2
+ import os
3
+ import torch
4
+ import re
5
+ from transformers import BertTokenizer, BertForSequenceClassification
6
+
7
  import os
8
 
9
  # Force Hugging Face to use /tmp as cache