LuxExistentia commited on
Commit
647618e
·
verified ·
1 Parent(s): d505611

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,12 +4,12 @@ import time
4
  from PIL import Image
5
  import os
6
 
7
- model_path = "deploying model/" + "vit_xsmall_patch16_clip_224(trainble_0.15) (Acc 98.44%, Loss 0.168152).onnx"
8
  input_size = [1, 3, 224, 224]
9
  img_size = input_size[-1]
10
 
11
  title = "Gender Vision mini"
12
- description = "An ViT(xsmall_clip) based model(fine tuned with Custom dataset : around 800 train images & 200 test iamges) Accuracy : around 98.4% with the custom test dataset. Optimized with ONNX(around 1.7 times faster than PyTorch version on cpu)"
13
  article = "Through bunch of fine tuning and experiments. !REMEMBER! This model can be wrong."
14
 
15
  def predict(img):
 
4
  from PIL import Image
5
  import os
6
 
7
+ model_path = "deploying model/" + "vit_xsmall_patch16_clip_224(trainble_0.00) (F1 Score 96.83%, Loss 0.229548).onnx"
8
  input_size = [1, 3, 224, 224]
9
  img_size = input_size[-1]
10
 
11
  title = "Gender Vision mini"
12
+ 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)"
13
  article = "Through bunch of fine tuning and experiments. !REMEMBER! This model can be wrong."
14
 
15
  def predict(img):