Update README.md
Browse files
README.md
CHANGED
|
@@ -78,8 +78,7 @@ predicted_class = torch.argmax(output, dim=1).item()
|
|
| 78 |
class_labels = {0: "Normal (Healthy)", 1: "Pneumonia"}
|
| 79 |
|
| 80 |
print(f"✅ Predicted Class: {class_labels.get(predicted_class, 'Unknown')}")
|
| 81 |
-
|
| 82 |
-
---
|
| 83 |
|
| 84 |
## **📊 Evaluation Results**
|
| 85 |
|
|
|
|
| 78 |
class_labels = {0: "Normal (Healthy)", 1: "Pneumonia"}
|
| 79 |
|
| 80 |
print(f"✅ Predicted Class: {class_labels.get(predicted_class, 'Unknown')}")
|
| 81 |
+
```
|
|
|
|
| 82 |
|
| 83 |
## **📊 Evaluation Results**
|
| 84 |
|