hyunseoki commited on
Commit
4a3b112
1 Parent(s): 435e3d0
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import torch
10
  model_name = 'hyunseoki/ReMoDetect-deberta'
11
 
12
  THESHOLD=4.0
13
- predictor = AutoModelForSequenceClassification.from_pretrained(model_name, force_download=True)
14
  tokenizer = AutoTokenizer.from_pretrained(model_name)
15
  predictor.eval()
16
 
 
10
  model_name = 'hyunseoki/ReMoDetect-deberta'
11
 
12
  THESHOLD=4.0
13
+ predictor = AutoModelForSequenceClassification.from_pretrained(model_name)
14
  tokenizer = AutoTokenizer.from_pretrained(model_name)
15
  predictor.eval()
16
 
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  torch
2
  numpy
3
- transformers==4.28.1
4
  datasets==2.12.0
5
  matplotlib
6
  tqdm
 
1
  torch
2
  numpy
3
+ transformers==4.44.2
4
  datasets==2.12.0
5
  matplotlib
6
  tqdm