Spaces:
Runtime error
Runtime error
Sadia commited on
Commit ·
4ceaa9f
1
Parent(s): ed20885
improved model trained on men and women images separately
Browse files- app.py +2 -2
- curlyhairmodel100percent.pkl → curlyhairmodelv2.pkl +0 -0
- curlyhairmodelv3.pkl +3 -0
app.py
CHANGED
|
@@ -2,7 +2,7 @@ import gradio as gr
|
|
| 2 |
from fastai.vision.all import *
|
| 3 |
import skimage
|
| 4 |
|
| 5 |
-
learn = load_learner('
|
| 6 |
|
| 7 |
categories = ('Curly', 'Straight')
|
| 8 |
|
|
@@ -27,7 +27,7 @@ enable_queue=True
|
|
| 27 |
|
| 28 |
gr.Interface(
|
| 29 |
fn=predict,
|
| 30 |
-
inputs=gr.inputs.Image(shape=(
|
| 31 |
outputs=gr.outputs.Label(),
|
| 32 |
title=title,
|
| 33 |
description=description,
|
|
|
|
| 2 |
from fastai.vision.all import *
|
| 3 |
import skimage
|
| 4 |
|
| 5 |
+
learn = load_learner('curlyhairmodelv3.pkl')
|
| 6 |
|
| 7 |
categories = ('Curly', 'Straight')
|
| 8 |
|
|
|
|
| 27 |
|
| 28 |
gr.Interface(
|
| 29 |
fn=predict,
|
| 30 |
+
inputs=gr.inputs.Image(shape=(128, 128)),
|
| 31 |
outputs=gr.outputs.Label(),
|
| 32 |
title=title,
|
| 33 |
description=description,
|
curlyhairmodel100percent.pkl → curlyhairmodelv2.pkl
RENAMED
|
File without changes
|
curlyhairmodelv3.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d72239f63dd01f18c9850a27c0175d7df1136d8d121079a21b70e49fdb777afa
|
| 3 |
+
size 46958305
|