UserPollo
/

moire-pattern-detector

Image Classification

moire-detection

screen-detection

digital-forensics

Model card Files Files and versions

Update critical error of README.md

#1

by JvL359 - opened 19 days ago

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -146,5 +146,5 @@ with torch.no_grad():
     probs = torch.softmax(logits, dim=1)
     conf, pred = probs.max(dim=1)
-print(f"Prediction: {classes[str(pred.item())]} (Confidence: {conf.item()*100:.1f}%)")
 ```

     probs = torch.softmax(logits, dim=1)
     conf, pred = probs.max(dim=1)
+print(f"Prediction: {classes[int(pred.item())]} (Confidence: {conf.item()*100:.1f}%)")
 ```