pratapaadii
/

image_classification

Image Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

pratapaadii commited on Jun 3, 2024

Commit

37c2774

·

verified ·

1 Parent(s): cccd535

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -4,7 +4,6 @@ base_model: google/vit-base-patch16-224-in21k
 tags:
 - generated_from_trainer
 datasets:
-- imagefolder
 - FastJobs/Visual_Emotional_Analysis
 metrics:
 - accuracy
@@ -31,7 +30,7 @@ should probably proofread and complete it, then remove this comment. -->
 # image_classification
-This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
 - Loss: 1.1877
 - Accuracy: 0.6

 tags:
 - generated_from_trainer
 datasets:
 - FastJobs/Visual_Emotional_Analysis
 metrics:
 - accuracy
 # image_classification
+This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the [FastJobs/Visual_Emotional_Analysis](https://huggingface.co/datasets/FastJobs/Visual_Emotional_Analysis) dataset.
 It achieves the following results on the evaluation set:
 - Loss: 1.1877
 - Accuracy: 0.6