Update app.py

app.py

@@ -80,7 +80,7 @@ class TextClassificationApp:
         CONV_FILTERS = [32, 32]  # Number of filters for each kernel size (4 and 5)
         BIDIRECTIONAL_UNITS = 128
         self.model = BCNN(EMBEDDING_DIM, OUTPUT_DIM, DROPOUT, BIDIRECTIONAL_UNITS, CONV_FILTERS)
-        self.model = torch.load(r'toxic.pt',map_location=torch.device('cpu'))
+        self.model = torch.load(r'toxic.pt',map_location=torch.device('cpu'),weights_only = False)
         self.model.eval()  # Set to evaluation mode
         
         # Store class names