vaniv commited on
Commit
2651551
·
verified ·
1 Parent(s): cf251ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def get_outfit_recommendation(pred_class):
52
  elif pred_class == 'Bag':
53
  return 'Top'
54
  else:
55
- return 'Top'
56
 
57
  def predict(image):
58
  pred_class, pred_idx, outputs = learn.predict(image)
 
52
  elif pred_class == 'Bag':
53
  return 'Top'
54
  else:
55
+ return 'Item'
56
 
57
  def predict(image):
58
  pred_class, pred_idx, outputs = learn.predict(image)