jfrery-zama commited on
Commit
d4d1e29
·
unverified ·
1 Parent(s): fd75825
Files changed (1) hide show
  1. wasm-demo.js +7 -20
wasm-demo.js CHANGED
@@ -500,32 +500,19 @@ $('btnDecrypt').onclick = () => {
500
  console.log('[Main] Decryption successful');
501
  console.log(`[Main] Result - flag: ${flag}, raw_score: ${rawScore}, total_g: ${total_g}`);
502
 
503
- // Convert raw score to watermark probability
504
- // rawScore: 0.5 = not watermarked, 1.0 = watermarked
505
- // Transform to: 0% = not watermarked, 100% = watermarked
506
- const watermarkProbability = Math.max(0, Math.min(1, 2 * (rawScore - 0.5)));
507
- const probabilityPercent = watermarkProbability * 100;
508
 
509
- // Determine result based on watermark probability ranges
510
- let resultText, probabilityText, resultClass;
511
-
512
- if (probabilityPercent < 30) {
513
- resultText = '👤 Likely Human-Generated';
514
- probabilityText = `Probability of being watermarked: ${probabilityPercent.toFixed(1)}%`;
515
- resultClass = 'inconclusive';
516
- } else if (probabilityPercent >= 30 && probabilityPercent < 70) {
517
- resultText = '⚠️ Uncertain';
518
- probabilityText = `Probability of being watermarked: ${probabilityPercent.toFixed(1)}%`;
519
- resultClass = 'inconclusive';
520
- } else {
521
- resultText = '✅ Likely AI-Generated (Watermarked)';
522
- probabilityText = `Probability of being watermarked: ${probabilityPercent.toFixed(1)}%`;
523
  resultClass = 'watermarked';
 
 
 
524
  }
525
 
526
  $('decResult').innerHTML = `
527
  <div class="watermark-flag ${resultClass}">${resultText}</div>
528
- <div class="watermark-score">${probabilityText}</div>
529
  `;
530
  } catch (e) {
531
  console.error('[Main] Decryption error:', e);
 
500
  console.log('[Main] Decryption successful');
501
  console.log(`[Main] Result - flag: ${flag}, raw_score: ${rawScore}, total_g: ${total_g}`);
502
 
503
+ // Determine result based on flag only
504
+ let resultText, resultClass;
 
 
 
505
 
506
+ if (flag === 1) {
507
+ resultText = '✅ AI-Generated (Watermarked)';
 
 
 
 
 
 
 
 
 
 
 
 
508
  resultClass = 'watermarked';
509
+ } else {
510
+ resultText = '👤 Human-Generated';
511
+ resultClass = 'inconclusive';
512
  }
513
 
514
  $('decResult').innerHTML = `
515
  <div class="watermark-flag ${resultClass}">${resultText}</div>
 
516
  `;
517
  } catch (e) {
518
  console.error('[Main] Decryption error:', e);