Update README.md

README.md

@@ -1,16 +1,26 @@
 ---
 license: mit
+pipeline_tag: image-classification
 ---
 
-A classification using mmpretrain trained to classify western images.
+A classification using mmpretrain trained to classify western images based on ConvNeXtV2-tiny.Used for classifying anime images based on whether they are in the Western style.
 
+The evaluation accuracy on the validation set is 95%.
+
+Trained using 7,000 Western images and 8,000 non-Western images, with the Western training set sampled from e-hentai.
+
+Of course, this model also has many shortcomings, such as a very low recognition accuracy for line-drawing images.
+
+# How to use
+
+Python>=3.9
 
 ```
-testimgs/ 
-├── western 
-│   ├── test1.png 
-│   └── ... 
-└── not_western 
-    ├── test2.png 
-    └── ... 
+Install pytorch
+
+pip3 install openmim
+
+edit infer.py , change "path = './testimg/'" to your target folder
+
+python infer.py
 ```