Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
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)
|
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):
|