rashmi commited on
Commit
b57be36
1 Parent(s): 1553854

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ predictor = DefaultPredictor(cfg)
32
 
33
 
34
  def inference(img):
35
- class_names = ['astro', 'cort', 'sh-sy5y']
36
  im = np.asarray(Image.open(img).convert('RGB'))
37
  outputs = predictor(im)
38
  pred_classes = outputs['instances'].pred_classes.cpu().numpy().tolist()
 
32
 
33
 
34
  def inference(img):
35
+ class_names = ['0','1','2'] #['astro', 'cort', 'sh-sy5y']
36
  im = np.asarray(Image.open(img).convert('RGB'))
37
  outputs = predictor(im)
38
  pred_classes = outputs['instances'].pred_classes.cpu().numpy().tolist()