YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)
Lung Cancer Classifier – ResNet50 (IQ-OTHNCCD Dataset)
This model is a fine-tuned ResNet50 trained on the IQ-OTHNCCD Lung Cancer dataset. Classes:
- Bengin cases
- Malignant cases
- Normal cases
How to use (PyTorch example)
from huggingface_hub import hf_hub_download
import torch
from torchvision import models, transforms
from PIL import Image
ckpt = hf_hub_download(repo_id="Sarth001/LungCanver", filename="best_model_v2_fixed.pth")
state = torch.load(ckpt, map_location='cpu')["model_state_dict"]
model = models.resnet50(pretrained=False)
model.fc = torch.nn.Linear(model.fc.in_features, 3)
model.load_state_dict(state, strict=False)
model.eval()
Disclaimer
For research/educational use only. Not for clinical diagnosis.
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support