Update app.py

app.py

@@ -35,12 +35,12 @@ cfg.MODEL.DEVICE= 'cpu'  # move to cpu
 predictor = DefaultPredictor(cfg)  # create model
 
 # -- load design modernity model for classification --------------------------------------------------------------
-DesignModernityModel = torch.load("DesignModernityModel.pt")
+DesignModernityModel = torch.load("DesignModernityModelBonus.pt")
 
 DesignModernityModel.eval() # set state of the model to inference
 
 # Set class labels
-LABELS = ['2000-2003', '2006-2008', '2009-2011', '2012-2014', '2015-2018']
+LABELS = ['2000-2003', '2004-2006', '2007-2009', '2010-2012', '2013-2015', '2016-2019']
 n_labels = len(LABELS)
 
 # define maéan and std dev for normalization