Update app.py

app.py

@@ -17,7 +17,7 @@ else:
 tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
 
 # Load model
-model = CommentClassificationModel(config_path="config.json")
+model = CommentMTLModel(config_path="config.json")
 model.load_state_dict(torch.load("pytorch_model.bin", map_location=device))
 model.to(device)
 model.eval()