Zetatech
/

pvt-small-224

Image Classification

Model card Files Files and versions

Xrenya commited on May 28, 2023

Commit

780b0d2

·

1 Parent(s): 8ab3048

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -45,8 +45,8 @@ import requests
 url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
 image = Image.open(requests.get(url, stream=True).raw)
-processor = PvtImageProcessor.from_pretrained('Xrenya/pvt-small-224')
-model = PvtForImageClassification.from_pretrained('Xrenya/pvt-small-224')
 inputs = processor(images=image, return_tensors="pt")
 outputs = model(**inputs)

 url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
 image = Image.open(requests.get(url, stream=True).raw)
+processor = PvtImageProcessor.from_pretrained('Zetatech/pvt-small-224')
+model = PvtForImageClassification.from_pretrained('Zetatech/pvt-small-224')
 inputs = processor(images=image, return_tensors="pt")
 outputs = model(**inputs)