Dzeniks
/

roberta-fact-check

Text Classification

Model card Files Files and versions

Update README.md

#2

by SameerAhmed-7 - opened Dec 20, 2024

base: refs/heads/main

←

from: refs/pr/2

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -41,7 +41,7 @@ model.eval()
 with torch.no_grad():
   prediction = model(**x)
-label = torch.argmax(outputs[0]).item()
 print(f"Label: {label}")
 ```

 with torch.no_grad():
   prediction = model(**x)
+label = torch.argmax(prediction[0]).item()
 print(f"Label: {label}")
 ```