Update README.md
Browse files
README.md
CHANGED
|
@@ -82,7 +82,7 @@ predicted_class = model.config.id2label[predicted_class_id]
|
|
| 82 |
|
| 83 |
print(f"Predicted class: {predicted_class}")
|
| 84 |
for i, score in enumerate(probs[0]):
|
| 85 |
-
print(f"{model.config.id2label[i]}: {score:.
|
| 86 |
```
|
| 87 |
|
| 88 |
---
|
|
|
|
| 82 |
|
| 83 |
print(f"Predicted class: {predicted_class}")
|
| 84 |
for i, score in enumerate(probs[0]):
|
| 85 |
+
print(f"{model.config.id2label[i]}: {score:.8f}")
|
| 86 |
```
|
| 87 |
|
| 88 |
---
|