Update README.md
Browse files
README.md
CHANGED
@@ -16,7 +16,7 @@ from transformers import ResNetForImageClassification
|
|
16 |
|
17 |
processor = AutoImageProcessor.from_pretrained("glazzova/body_type")
|
18 |
model = ResNetForImageClassification.from_pretrained('glazzova/body_type')
|
19 |
-
image = Image.open('
|
20 |
inputs = processor(image, return_tensors="pt")
|
21 |
|
22 |
with torch.no_grad():
|
|
|
16 |
|
17 |
processor = AutoImageProcessor.from_pretrained("glazzova/body_type")
|
18 |
model = ResNetForImageClassification.from_pretrained('glazzova/body_type')
|
19 |
+
image = Image.open('your_photo.jpg')
|
20 |
inputs = processor(image, return_tensors="pt")
|
21 |
|
22 |
with torch.no_grad():
|