glazzova
/

body_type

Image Classification

Inference Endpoints

Model card Files Files and versions Community

glazzova commited on May 12

Commit

356ffe5

•

1 Parent(s): 1979237

Update README.md

Files changed (1) hide show

README.md +3 -2

README.md CHANGED Viewed

@@ -11,12 +11,13 @@ pipeline_tag: image-classification
 ### Запуск модели
 ```python
 from PIL import Image
-from transformers import ResNetForImageClassification
 processor = AutoImageProcessor.from_pretrained("glazzova/body_type")
 model = ResNetForImageClassification.from_pretrained('glazzova/body_type')
-image = Image.open('your_photo.jpg')
 inputs = processor(image, return_tensors="pt")
 with torch.no_grad():

 ### Запуск модели
 ```python
+import torch
 from PIL import Image
+from transformers import ResNetForImageClassification, AutoImageProcessor
 processor = AutoImageProcessor.from_pretrained("glazzova/body_type")
 model = ResNetForImageClassification.from_pretrained('glazzova/body_type')
+image = Image.open('/Users/swag2004/Downloads/photo_2024-05-11 16.17.08.jpeg')
 inputs = processor(image, return_tensors="pt")
 with torch.no_grad():