Spaces:
Runtime error
Runtime error
Commit ·
b03550c
1
Parent(s): 0b0cae3
update to watches learner
Browse files- app.py +4 -4
- auctionet-model.pkl +2 -2
app.py
CHANGED
|
@@ -1,10 +1,10 @@
|
|
| 1 |
-
import timm
|
| 2 |
import gradio as gr
|
| 3 |
from fastai.vision.all import *
|
| 4 |
import skimage
|
| 5 |
from PIL import Image
|
|
|
|
| 6 |
|
| 7 |
-
learn = load_learner('
|
| 8 |
labels = learn.dls.vocab
|
| 9 |
|
| 10 |
def classify_image(img):
|
|
@@ -13,5 +13,5 @@ def classify_image(img):
|
|
| 13 |
return dict(zip(labels, map(float, probs)))
|
| 14 |
|
| 15 |
|
| 16 |
-
iface = gr.Interface(fn=classify_image, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=3), interpretation="default", title="
|
| 17 |
-
iface.launch(inline=False)
|
|
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
from fastai.vision.all import *
|
| 3 |
import skimage
|
| 4 |
from PIL import Image
|
| 5 |
+
from timm import *
|
| 6 |
|
| 7 |
+
learn = load_learner('auctionet-model.pkl')
|
| 8 |
labels = learn.dls.vocab
|
| 9 |
|
| 10 |
def classify_image(img):
|
|
|
|
| 13 |
return dict(zip(labels, map(float, probs)))
|
| 14 |
|
| 15 |
|
| 16 |
+
iface = gr.Interface(fn=classify_image, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=3), interpretation="default", title="Watches classifier")
|
| 17 |
+
iface.launch(share=True,inline=False)
|
auctionet-model.pkl
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5bafeb7db184cbe019dfd7a421c03dd9de2c3892f86a228c4ce173439dfb7839
|
| 3 |
+
size 114682841
|