Update app.py

app.py

@@ -47,7 +47,7 @@ class MultiTaskBertModel(torch.nn.Module):
         
         return model
 
-model = MultiTaskBertModel.from_pretrained("DmitriySv/ticket_classifer")
+model = MultiTaskBertModel.from_pretrained("DmitriySv/ticket_classifer", 28, 3)
 tokenizer = BertTokenizer.from_pretrained("DmitriySv/ticket_classifer")
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')