abhilash88
/

age-gender-prediction

Image Classification

vision-transformer

gender-classification

computer-vision

multi-task-learning

Model card Files Files and versions

abhilash88 commited on Aug 14

Commit

5b22c19

·

verified ·

1 Parent(s): 1b298f7

Update config.json

Files changed (1) hide show

config.json +1 -12

config.json CHANGED Viewed

@@ -12,18 +12,7 @@
     "AutoConfig": "model.AgeGenderConfig",
     "AutoModelForImageClassification": "model.AgeGenderViTModel"
   },
-  "custom_pipelines": {
-    "image-classification": {
-      "impl": "model.AgeGenderImageClassificationPipeline",
-      "pt": ["AgeGenderViTModel"]
-    }
-  },
   "pipeline_tag": "image-classification",
   "torch_dtype": "float32",
-  "transformers_version": "4.36.0",
-  "task_specific_params": {
-    "image-classification": {
-      "top_k": 1
-    }
-  }
 }

     "AutoConfig": "model.AgeGenderConfig",
     "AutoModelForImageClassification": "model.AgeGenderViTModel"
   },
   "pipeline_tag": "image-classification",
   "torch_dtype": "float32",
+  "transformers_version": "4.36.0"
 }