Spaces:

swangfr
/

Amazon-product-image-classification

Sleeping

swangfr commited on Mar 22

Commit

ef1aff3

•

1 Parent(s): 58e5cc9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -63,8 +63,9 @@ if file_name is not None:
         'Xbox One',
         'Clothing, Shoes & Jewelry ',
         'Home & Kitchen ',
-        'Toys & Games ',
-        'Video Games ']])
     multilabel.inverse_transform(prediction.reshape(1,-1))

         'Xbox One',
         'Clothing, Shoes & Jewelry ',
         'Home & Kitchen ',
+        'Video Games '
+        'Toys & Games '
+        ]])
     multilabel.inverse_transform(prediction.reshape(1,-1))