codewithdark
/

vit-chest-xray

Image Classification

Model card Files Files and versions

codewithdark commited on Jan 23, 2025

Commit

bfa790c

·

verified ·

1 Parent(s): 94d22eb

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -59,8 +59,8 @@ import torch
 from PIL import Image
 # Load model and feature extractor
-model = ViTForImageClassification.from_pretrained('codewithdark/chest-xray-classifier')
-feature_extractor = ViTFeatureExtractor.from_pretrained('codewithdark/chest-xray-classifier')
 # Prepare an image for prediction
 image = Image.open('path_to_chest_xray_image.jpg')

 from PIL import Image
 # Load model and feature extractor
+model = ViTForImageClassification.from_pretrained('codewithdark/vit-chest-xray')
+feature_extractor = ViTFeatureExtractor.from_pretrained('codewithdark/vit-chest-xray')
 # Prepare an image for prediction
 image = Image.open('path_to_chest_xray_image.jpg')