Update application file

app.py

@@ -68,6 +68,17 @@ def tnet(inputs, num_features):
 EXAMPLES_PATH = './examples'
 model = from_pretrained_keras('keras-io/PointNet')
 
+CLASS_MAP = {0: 'chair',
+ 1: 'sofa',
+ 2: 'desk',
+ 3: 'bed',
+ 4: 'dresser',
+ 5: 'night_stand',
+ 6: 'toilet',
+ 7: 'bathtub',
+ 8: 'monitor',
+ 9: 'table'}
+
 def infer(img_path):
     mesh = trimesh.load(img_path.name)
     points = mesh.sample(2048)