d-fuentep commited on
Commit
cff48d2
·
verified ·
1 Parent(s): f00ed78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -10,7 +10,9 @@ learn = load_learner(model_path)
10
  def predict(image):
11
  pred, _, probs = learn.predict(image)
12
  return {learn.dls.vocab[i]: float(probs[i]) for i in range(len(probs))}
13
-
 
 
14
  #interface
15
- demo = gr.Interface(fn=predict, inputs="image", outputs="label")
16
  demo.launch()
 
10
  def predict(image):
11
  pred, _, probs = learn.predict(image)
12
  return {learn.dls.vocab[i]: float(probs[i]) for i in range(len(probs))}
13
+ title = "Outfit classifier"
14
+ description = "Are you ready for that dinner? I'll tell you whether your outfit is formal or streetwear (do not take me seriously) "
15
+ examples = ['formal', 'streetwear']
16
  #interface
17
+ demo = gr.Interface(fn=predict,inputs="image", outputs="label",title=title,description=description,examples=examples)
18
  demo.launch()