mc0c0z commited on
Commit
6ad4337
·
verified ·
1 Parent(s): c6cc6e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from transformers import MBartForConditionalGeneration, AutoTokenizer, AutoModel
10
  from torch import nn
11
  import torch.nn.functional as F
12
  from underthesea import word_tokenize
13
-
14
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
15
 
16
  # Load multi task model
 
10
  from torch import nn
11
  import torch.nn.functional as F
12
  from underthesea import word_tokenize
13
+
14
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
15
 
16
  # Load multi task model