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