yaful commited on
Commit
a5a9283
1 Parent(s): 7897a5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ model_dir = "nealcly/detection-longformer"
15
  tokenizer = AutoTokenizer.from_pretrained(model_dir)
16
  model = AutoModelForSequenceClassification.from_pretrained(model_dir).to(device)
17
 
18
- def detect(input_text,device,th=-3.08583984375):
19
  label2decisions = {
20
  0: "machine-generated",
21
  1: "human-written",
 
15
  tokenizer = AutoTokenizer.from_pretrained(model_dir)
16
  model = AutoModelForSequenceClassification.from_pretrained(model_dir).to(device)
17
 
18
+ def detect(input_text,th=-3.08583984375):
19
  label2decisions = {
20
  0: "machine-generated",
21
  1: "human-written",