Eraly-ml commited on
Commit
e24cb10
Β·
verified Β·
1 Parent(s): 2bb7332

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
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
- ![Training Metrics](https://huggingface.co/Eraly-ml/Skin-AI/edit/main/plot.jpg)
47
 
48
  - **Accuracy & Precision improved with hyperparameter tuning**
49
  - **Training loss consistently decreased, showing model convergence**
@@ -56,7 +56,7 @@ from torchvision import transforms
56
  from PIL import Image
57
 
58
  # Load the model
59
- model = torch.jit.load("medconvnext_scripted.pt")
60
  model.eval()
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:** [YOUR_HF_SPACE_URL]
87
- **πŸ“Œ Author:** [Your Name]
88
 
89
 
90
  My telegram @eralyf
 
43
  ## πŸš€ Results
44
  Below are key performance graphs from TensorBoard:
45
 
46
+ ![Training Metrics](plot.jpg)
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