codewithdark
/

vit-chest-xray

Image Classification

Inference Endpoints

Model card Files Files and versions Community

codewithdark commited on 18 days ago

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')