Spaces:
Runtime error
Runtime error
Commit
·
75434eb
1
Parent(s):
676ab2f
Update app.py
Browse files
app.py
CHANGED
@@ -64,8 +64,8 @@ def predict(query):
|
|
64 |
logits = model(tokens.to(device), attention_mask=mask.to(device))[0]
|
65 |
probs = logits.softmax(dim=-1)
|
66 |
|
67 |
-
|
68 |
-
fake, real = probs.detach().cpu().flatten().numpy().tolist() # PirateXX/AI-Content-Detector-V2
|
69 |
|
70 |
return real
|
71 |
|
|
|
64 |
logits = model(tokens.to(device), attention_mask=mask.to(device))[0]
|
65 |
probs = logits.softmax(dim=-1)
|
66 |
|
67 |
+
real, fake = probs.detach().cpu().flatten().numpy().tolist() # Hello-SimpleAI/chatgpt-detector-roberta
|
68 |
+
# fake, real = probs.detach().cpu().flatten().numpy().tolist() # PirateXX/AI-Content-Detector-V2
|
69 |
|
70 |
return real
|
71 |
|