Saad0KH commited on
Commit
f3b4b4b
Β·
verified Β·
1 Parent(s): 5a7f292

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def root():
33
  def classify():
34
  data = request.json
35
  print(data)
36
- clothes = data['clothes']
37
  image = data['image']
38
  result = segment_image(clothes, image)
39
  return jsonify({'result': encode_image_to_base64(result)})
 
33
  def classify():
34
  data = request.json
35
  print(data)
36
+ clothes = ["Hat", "Upper-clothes", "Skirt", "Pants", "Dress", "Belt", "Left-shoe", "Right-shoe", "Scarf"]
37
  image = data['image']
38
  result = segment_image(clothes, image)
39
  return jsonify({'result': encode_image_to_base64(result)})