hoxmark commited on
Commit
3cfa45a
1 Parent(s): b90e60f
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -9,9 +9,8 @@ def predict(img):
9
  pred,pred_idx,probs = learn.predict(img)
10
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
11
 
12
- import gradio as gro
13
 
14
- title = "Pet Breed Classifier"
15
  description = "A bunads classifier trained to classify if a bunad is a Nordlandsbunad, Trønderbunad, Hardangerbunaden, Vestfoldbunad"
16
 
17
  gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=4)).launch(share=True)
 
9
  pred,pred_idx,probs = learn.predict(img)
10
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
11
 
 
12
 
13
+ title = "Norwegian Bunads Classifier"
14
  description = "A bunads classifier trained to classify if a bunad is a Nordlandsbunad, Trønderbunad, Hardangerbunaden, Vestfoldbunad"
15
 
16
  gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=4)).launch(share=True)