LuxExistentia commited on
Commit
3cb2603
·
1 Parent(s): de89970

Changed model from Vit Xsmall clip to resnet50_clip_gap. Changed descriptioin and .onnx file

Browse files
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ .ipynb_checkpoints/
app.py CHANGED
@@ -10,7 +10,7 @@ input_size = [1, 3, 224, 224]
10
  img_size = input_size[-1]
11
 
12
  title = "Gender Vision mini"
13
- description = "An ViT(xsmall_clip) based model(fine tuned with Custom dataset : around 800 train images & 200 test iamges) F1 Score : around 96.8%(0.9684) with the custom test dataset. Optimized with ONNX(around 1.7 times faster than PyTorch version on cpu)"
14
  article = "Through bunch of fine tuning and experiments. !REMEMBER! This model can be wrong."
15
 
16
  def predict(img):
 
10
  img_size = input_size[-1]
11
 
12
  title = "Gender Vision mini"
13
+ description = "An resnet50_clip_gap based model(fine tuned with Custom dataset : around 800 train images & 200 test iamges) F1 Score : 100%(1.00) with the custom test dataset. Optimized with ONNX(around 1.7 times faster than PyTorch version on cpu)"
14
  article = "Through bunch of fine tuning and experiments. !REMEMBER! This model can be wrong."
15
 
16
  def predict(img):
deploying model/{vit_xsmall_patch16_clip_224(trainble_0.00) (F1 Score 96.83%, Loss 0.229548).onnx → resnet50_clip_gap(trainble_0.00) (F1 Score 100.00%, Loss 0.149292).onnx} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:51b816317383367efa342177f42718563df644d1cbe616d7023924b007fef7c8
3
- size 32699557
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5bbf7a95eda37fe6b85764eac8d0796221e07f92c7cfcff91ee779d5cd3b833
3
+ size 94052414