Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import torch
 from torchvision import models, transforms
 from PIL import Image
-'''
+
 # -- get torch and cuda version
 TORCH_VERSION = ".".join(torch.__version__.split(".")[:2])
 CUDA_VERSION = torch.__version__.split("+")[-1]
@@ -38,7 +38,7 @@ outputs = predictor(im)
 
 print(outputs["instances"].pred_classes)
 print(outputs["instances"].pred_boxes)
-'''
+
 # -- load Mask R-CNN model for segmentation
 DesignModernityModel = torch.load("DesignModernityModel.pt")
 
@@ -47,7 +47,7 @@ DesignModernityModel = torch.load("DesignModernityModel.pt")
 
 DesignModernityModel.eval() # set state of the model to inference
 
-LABELS = ['2000-2004', '2006-2008', '2009-2011', '2012-2015', '2016-2018']
+LABELS = ['2000-2003', '2006-2008', '2009-2011', '2012-2014', '2015-2018']
 n_labels = len(LABELS)
 
 MEAN = [0.485, 0.456, 0.406]