DmitriySv commited on
Commit
ab777d7
·
verified ·
1 Parent(s): 30108a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,8 +48,8 @@ class MultiTaskBertModel(torch.nn.Module):
48
 
49
  return model
50
 
51
- model = MultiTaskBertModel.from_pretrained("DmitriySv/ticket_classifer/tree/main", 28, 3)
52
- tokenizer = BertTokenizer.from_pretrained("DmitriySv/ticket_classifer/tree/main")
53
 
54
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
55
  model = model.to(device)
 
48
 
49
  return model
50
 
51
+ model = MultiTaskBertModel.from_pretrained("DmitriySv/ticket_classifer/", 28, 3)
52
+ tokenizer = BertTokenizer.from_pretrained("DmitriySv/ticket_classifer/")
53
 
54
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
55
  model = model.to(device)