Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -23,9 +23,9 @@ paraphrase_model = paraphrase_model.to(device)
|
|
| 23 |
similarity_model = SentenceTransformer('all-MiniLM-L6-v2')
|
| 24 |
|
| 25 |
# ----------------------
|
| 26 |
-
# Local AI Detector (
|
| 27 |
# ----------------------
|
| 28 |
-
AI_DETECTOR_MODEL = "
|
| 29 |
ai_detector = pipeline("text-classification", model=AI_DETECTOR_MODEL, device=0 if torch.cuda.is_available() else -1)
|
| 30 |
|
| 31 |
# ----------------------
|
|
|
|
| 23 |
similarity_model = SentenceTransformer('all-MiniLM-L6-v2')
|
| 24 |
|
| 25 |
# ----------------------
|
| 26 |
+
# Local AI Detector (roberta-base-openai-detector)
|
| 27 |
# ----------------------
|
| 28 |
+
AI_DETECTOR_MODEL = "roberta-base-openai-detector"
|
| 29 |
ai_detector = pipeline("text-classification", model=AI_DETECTOR_MODEL, device=0 if torch.cuda.is_available() else -1)
|
| 30 |
|
| 31 |
# ----------------------
|