jaimin commited on
Commit
3eb6e90
β€’
1 Parent(s): 72beadd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -8,6 +8,9 @@ import pytorch_lightning as pl
8
  import torch
9
  import itertools
10
  import random
 
 
 
11
 
12
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
13
 
 
8
  import torch
9
  import itertools
10
  import random
11
+ import nltk
12
+ from nltk.tokenize import sent_tokenize
13
+ nltk.download('punkt')
14
 
15
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
16