Hugging Face's logo

DineshKumar1329
/
DogCat_Classifier

Transformers
English
resnet-18
code
Model card Files Community
DineshKumar1329 commited on
Commit
522791f
·
verified ·
1 Parent(s): 1e171fb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -33,11 +33,16 @@ This repository contains a ResNet-based convolutional neural network trained to
33
  - Accuracy: 90.27% on test images
34
  - Training Time: Approximately 1 hour on NVIDIA GTX 1080 Ti
35
 
 
 
 
36
  ## Usage
37
 
38
  ### Installation:
39
  - Dependencies: PyTorch, TorchVision, matplotlib
40
 
 
 
41
  ### Inference:
42
  ```python
43
  import torch
@@ -175,5 +180,4 @@ plot_image(image_path, prediction)
175
  The predicted class for the image is: Cat
176
 
177
 
178
- ![image/png](https://cdn-uploads.huggingface.co/production/uploads/65dd9dc387458446d0a9da32/IwtLTneWqyRz1uYuYY1Kp.png)
179
 
 
33
  - Accuracy: 90.27% on test images
34
  - Training Time: Approximately 1 hour on NVIDIA GTX 1080 Ti
35
 
36
+
37
+ ## Results:
38
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/65dd9dc387458446d0a9da32/IwtLTneWqyRz1uYuYY1Kp.png)
39
  ## Usage
40
 
41
  ### Installation:
42
  - Dependencies: PyTorch, TorchVision, matplotlib
43
 
44
+
45
+
46
  ### Inference:
47
  ```python
48
  import torch
 
180
  The predicted class for the image is: Cat
181
 
182
 
 
183