taha092 commited on
Commit
fd8896a
·
verified ·
1 Parent(s): 208bdb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 (desklib/ai-text-detector-v1.01)
27
  # ----------------------
28
- AI_DETECTOR_MODEL = "desklib/ai-text-detector-v1.01"
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
  # ----------------------