Hugging Face's logo

parneetsingh022
/
dog-cat-classification

Transformers
Safetensors
custom
Inference Endpoints
Model card Files Community
parneetsingh022 commited on
Commit
9fb8739
•
1 Parent(s): 330d92f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -28,8 +28,8 @@ import torch
28
  from PIL import Image
29
  from torchvision import transforms
30
 
31
- from catdog_classifier.configuration import CustomModelConfig
32
- from catdog_classifier.model import CustomClassifier
33
  ```
34
 
35
  Loading the model:
 
28
  from PIL import Image
29
  from torchvision import transforms
30
 
31
+ from custom_classifier.configuration import CustomModelConfig
32
+ from custom_classifier.model import CustomClassifier
33
  ```
34
 
35
  Loading the model: