Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,10 +6,14 @@ from tokenizers import normalizers
|
|
| 6 |
from tokenizers.normalizers import Sequence, Replace, Strip
|
| 7 |
from tokenizers import Regex
|
| 8 |
|
| 9 |
-
|
| 10 |
|
| 11 |
-
|
| 12 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
|
| 14 |
model_2 = DebertaForSequenceClassification.from_pretrained("mihalykiss/best_merged_41_2", num_labels=41)
|
| 15 |
model_2.to(device).eval()
|
|
@@ -92,7 +96,7 @@ title = "AI Text Detector"
|
|
| 92 |
description = """
|
| 93 |
|
| 94 |
|
| 95 |
-
This tool uses
|
| 96 |
<br>
|
| 97 |
|
| 98 |
<div style="line-height: 1.8;">
|
|
|
|
| 6 |
from tokenizers.normalizers import Sequence, Replace, Strip
|
| 7 |
from tokenizers import Regex
|
| 8 |
|
| 9 |
+
from transformers import DebertaTokenizerFast
|
| 10 |
|
| 11 |
+
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 12 |
|
| 13 |
+
tokenizer = DebertaTokenizerFast.from_pretrained(
|
| 14 |
+
"microsoft/deberta-base",
|
| 15 |
+
add_prefix_space=True
|
| 16 |
+
)
|
| 17 |
|
| 18 |
model_2 = DebertaForSequenceClassification.from_pretrained("mihalykiss/best_merged_41_2", num_labels=41)
|
| 19 |
model_2.to(device).eval()
|
|
|
|
| 96 |
description = """
|
| 97 |
|
| 98 |
|
| 99 |
+
This tool uses a merged <b>DeBERTa</b> model to identify whether a given text was written by a human or generated by artificial intelligence (AI).
|
| 100 |
<br>
|
| 101 |
|
| 102 |
<div style="line-height: 1.8;">
|