albertramstedt commited on
Commit
b03550c
·
1 Parent(s): 0b0cae3

update to watches learner

Browse files
Files changed (2) hide show
  1. app.py +4 -4
  2. 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('european-ceramics.pkl')
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="Svensk keramik-AI")
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:1a4499481892159340e6035d1008e2496a53a540222bcba4c0783ea924a5cacb
3
- size 114667289
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5bafeb7db184cbe019dfd7a421c03dd9de2c3892f86a228c4ce173439dfb7839
3
+ size 114682841