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 |
# ----------------------
|