Xrenya
/

pvt-medium-224

Image Classification

Inference Endpoints

Model card Files Files and versions Community

Xrenya commited on May 28, 2023

Commit

346bf91

•

1 Parent(s): 3a96f3f

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-medium-224')
-model = PvtForImageClassification.from_pretrained('Xrenya/pvt-medium-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-medium-224')
+model = PvtForImageClassification.from_pretrained('Zetatech/pvt-medium-224')
 inputs = processor(images=image, return_tensors="pt")
 outputs = model(**inputs)