Cha1 commited on
Commit
f11a4fd
·
1 Parent(s): 1f713f9

description

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -10,4 +10,7 @@ def predict(img):
10
  pred,pred_idx,probs = learn.predict(img)
11
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
12
 
13
- gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=3)).launch(share=True)
 
 
 
 
10
  pred,pred_idx,probs = learn.predict(img)
11
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
12
 
13
+ gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=3)).launch(share=True)
14
+
15
+ title = "Lingerie brand classifier"
16
+ description = "This classifier can identify whether your lingerie is from Studio PIA, Atelier Bordelle or Edge O'Beyond. Trained w/ fast.ai and a bunch of pictures found w/ DuckDuckGo"