Update app.py

app.py

@@ -37,8 +37,8 @@ enc = OneHotEncoder(sparse=False)
 enc.fit(np.concatenate((train_yt, test_yt)).reshape((-1, 1)))
 # train_yv = enc.transform(self.train_yv.reshape((-1, 1))).astype(np.float64)
 # test_yv = enc.transform(self.test_yv.reshape((-1, 1))).astype(np.float64)
-train_yt = enc.transform(self.train_yt.reshape((-1, 1))).astype(np.float64)
-test_yt = enc.transform(self.test_yt.reshape((-1, 1))).astype(np.float64)
+train_yt = enc.transform(train_yt.reshape((-1, 1))).astype(np.float64)
+test_yt = enc.transform(test_yt.reshape((-1, 1))).astype(np.float64)
 
 # Map the image ids to the corresponding image URLs
 image_map_name = 'pascal_dataset.csv'