devloverumar commited on
Commit
75434eb
·
1 Parent(s): 676ab2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- # 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
 
 
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