Vignesh1997 commited on
Commit
90d6332
1 Parent(s): 5580efb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import time
6
  import json
7
  from typing import List
8
  import torch
 
9
  import random
10
  import logging
11
  from transformers import BertTokenizer, BertModel, BertConfig
 
6
  import json
7
  from typing import List
8
  import torch
9
+ from torch import nn
10
  import random
11
  import logging
12
  from transformers import BertTokenizer, BertModel, BertConfig