Update README.md
Browse files
README.md
CHANGED
|
@@ -43,7 +43,7 @@ The model was trained using PyTorch Lightning with automatic logging to TensorBo
|
|
| 43 |
## π Results
|
| 44 |
Below are key performance graphs from TensorBoard:
|
| 45 |
|
| 46 |
-

|
| 61 |
|
| 62 |
# Define image transformation
|
|
@@ -83,8 +83,8 @@ print("Predicted Class:", predicted_class)
|
|
| 83 |
This project is intended for research and educational purposes only. For clinical use, further validation is required.
|
| 84 |
|
| 85 |
---
|
| 86 |
-
**π Hugging Face Space:** [
|
| 87 |
-
**π Author:** [
|
| 88 |
|
| 89 |
|
| 90 |
My telegram @eralyf
|
|
|
|
| 43 |
## π Results
|
| 44 |
Below are key performance graphs from TensorBoard:
|
| 45 |
|
| 46 |
+

|
| 47 |
|
| 48 |
- **Accuracy & Precision improved with hyperparameter tuning**
|
| 49 |
- **Training loss consistently decreased, showing model convergence**
|
|
|
|
| 56 |
from PIL import Image
|
| 57 |
|
| 58 |
# Load the model
|
| 59 |
+
model = torch.jit.load("skinconvnext_scripted.pt")
|
| 60 |
model.eval()
|
| 61 |
|
| 62 |
# Define image transformation
|
|
|
|
| 83 |
This project is intended for research and educational purposes only. For clinical use, further validation is required.
|
| 84 |
|
| 85 |
---
|
| 86 |
+
**π Hugging Face Space:** [https://huggingface.co/spaces/Eraly-ml/Skin-AI]
|
| 87 |
+
**π Author:** [Eraly Gainulla]
|
| 88 |
|
| 89 |
|
| 90 |
My telegram @eralyf
|