Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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,
|
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",
|